numpy.ma.MaskedArray.compress

MaskedArray.compress(condition, axis=None, out=None) [source]

Возвращает a где условие True.

Если условие — MaskedArray, пропущенные значения рассматриваются как False.

Параметры:

condition : var Булево 1-мерный массив, выбирающий элементы для возврата. Если длина condition меньше размера a по оси, то вывод усекается до длины массива condition. axis : {None, int}, необязательно Ось, по которой выполняется операция. out : {None, ndarray}, необязательно Альтернативный массив вывода, в который помещается результат. Он должен иметь ту же форму, что и ожидаемый вывод, но тип будет преобразован при необходимости.

Возвращает:

result : MaskedArray Объект MaskedArray.

Примечания

Обратите внимание на разницу с compressed! Вывод compress имеет маску, а вывод compressed — нет.

Примеры

>>> x = np.ma.array([[1,2,3],[4,5,6],[7,8,9]], mask=[0] + [1,0]*4)
>>> print(x)
[[1 -- 3]
 [-- 5 --]
 [7 -- 9]]
>>> x.compress([1, 0, 1])
masked_array(data = [1 3],
      mask = [False False],
      fill_value=999999)

>>> x.compress([1, 0, 1], axis=1)
masked_array(data =
 [[1 3]
 [-- --]
 [7 9]],
      mask =
 [[False False]
 [ True  True]
 [False False]],
      fill_value=999999)