safe_mask

sklearn.utils.safe_mask(X, маска)

Возвращает маску, безопасную для использования на X.

Параметры:

X{array-like, sparse matrix}

Данные, к которым применяется маска.

маскаarray-like

Маска, применяемая к X.

Возвращает:

маскаndarray

Массив, который безопасно использовать на 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]])