numpy.ma.MaskedArray.compress

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

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

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

Параметры:

condition : var Булевый массив 1-d, выбирающий элементы для возврата. Если длина условия меньше размера a по оси, то выходные данные обрезаются до длины массива условия. 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)