sklearn.utils._safe_indexing?

sklearn.utils._safe_indexing(X, indices, *, axis=0

源碼

使用索引返回X的行,項目或列。

警告:

該實用程序已記錄,但是是私有的。 這意味著沒有任何棄用周期就可以破壞向后兼容性。

參數 說明
X array-like, sparse-matrix, list, pandas.DataFrame, pandas.Series
用于抽取行、項或列的數據。僅當axis=0時才支持列表。
indices bool, int, str, slice, array-like
如果axis = 0,則支持布爾型和整數數組,整數切片和標量整數。
如果axis = 1:要選擇單個列,對于所有X類型,索引可以是int類型,對于數據幀,只能是str。 除非X是稀疏矩陣,否則所選子集將是1D,在這種情況下它將是2D。
要選擇多列,索引可以是以下之一:列表,數組,切片。 這些容器中使用的類型可以是以下類型之一:int,“ bool”和str。 但是,僅當X是數據幀時才支持str。 所選的子集將是2D。
axis int, default=0
X將被沿其采樣的軸。 axis = 0將選擇行,而axis = 1將選擇列。
返回值 說明
subset 軸0或1上X的子集。

注:

支持CSR,CSC和LIL稀疏矩陣。 不支持COO稀疏矩陣。