安全遮罩

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]])