safe_mask

sklearn.utils.safe_mask(X, mask)

返回可安全用于 X 的掩码。

参数:

X{类数组对象, 稀疏矩阵}

应用于掩码的数据。

mask类数组对象

要在 X 上使用的掩码。

返回:

maskndarray

可以在 X 上安全使用的数组。

示例

>>> from sklearn.utils import safe_mask
>>> from scipy.sparse import csr_matrix
>>> data = csr_matrix([[1], [2], [3], [4], [5]])
>>> condition = [False, True, True, False, True]
>>> mask = safe_mask(data, condition)
>>> data[mask].toarray()
array([[2],
       [3],
       [5]])