numpy.ma.MaskedArray.compress

метод

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

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

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

Параметры:

conditionпеременная

Булевый массив 1-d, выбирающий, какие элементы возвращать. Если длина condition меньше размера a по оси, то выход усекается до длины массива condition.

axis{None, int}, необязательно

Ось, по которой выполняется операция.

out{None, ndarray}, необязательно

Альтернативный выходной массив, в который следует поместить результат. Он должен иметь такую же форму, как ожидаемый результат, но тип будет преобразован при необходимости.

Возвращаемое значение:

resultMaskedArray

Объект MaskedArray.

Примечания

Обратите внимание на различие с compressed! Результат compress имеет маску, результат compressed нет.

Примеры

>>> x = np.ma.array([[1,2,3],[4,5,6],[7,8,9]], mask=[0] + [1,0]*4)
>>> x
masked_array(
  data=[[1, --, 3],
        [--, 5, --],
        [7, --, 9]],
  mask=[[False,  True, False],
        [ True, False,  True],
        [False,  True, False]],
  fill_value=999999)
>>> 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)