Python numpy MaskedArray.compress用法及代码示例

本文简要介绍 python 语言中 numpy.ma.MaskedArray.compress 的用法。

用法:

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

返回a其中条件是True.

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

参数：

condition： var

布尔一维数组，选择要返回的条目。如果 len(condition) 小于 a 沿轴的大小，则输出被截断为条件数组的长度。

axis： {无，整数}，可选

必须沿其执行操作的轴。

out： {无，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)
>>> 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)