numpy.ma.MaskedArray.compress

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

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

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

Параметры:

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

Возвращает:

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)