sklearn.utils
._safe_indexing¶

sklearn.utils.
_safe_indexing
(X, indices, axis=0)[source]¶ Return rows, items or columns of X using indices.
Warning
This utility is documented, but private. This means that backward compatibility might be broken without any deprecation cycle.
 Parameters
 Xarraylike, sparsematrix, list, pandas.DataFrame, pandas.Series
Data from which to sample rows, items or columns.
list
are only supported whenaxis=0
. indicesbool, int, str, slice, arraylike
If
axis=0
, boolean and integer arraylike, integer slice, and scalar integer are supported. If
axis=1
: to select a single column,
indices
can be ofint
type for allX
types andstr
only for dataframe. The selected subset will be 1D, unlessX
is a sparse matrix in which case it will be 2D.to select multiples columns,
indices
can be one of the following:list
,array
,slice
. The type used in these containers can be one of the following:int
, ‘bool’ andstr
. However,str
is only supported whenX
is a dataframe. The selected subset will be 2D.
 If
 axisint, default=0
The axis along which
X
will be subsampled.axis=0
will select rows whileaxis=1
will select columns.
 Returns
 subset
Subset of X on axis 0 or 1.
Notes
CSR, CSC, and LIL sparse matrices are supported. COO sparse matrices are not supported.