numpy.ma.MaskedArray.compress#

方法

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

返回 a，其中 condition 为 True。

如果 condition 是 MaskedArray，则缺失值被视为 False。

参数：:

conditionvar: 布尔型一维数组，用于选择要返回的条目。如果 len(condition) 小于 a 沿轴的大小，则输出将截断为 condition 数组的长度。
axis{None, int}, 可选: 必须执行操作的轴。
out{None, ndarray}, 可选: 用于放置结果的备用输出数组。它必须与预期输出具有相同的形状，但类型将在必要时进行强制转换。

返回：:

resultMaskedArray: 一个 MaskedArray 对象。

备注

请注意与 compressed 的区别！compress 的输出具有掩码，compressed 的输出没有。

示例

>>> import numpy as np
>>> 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)