Python numpy ma.where用法及代码示例

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

用法:

ma.where(condition, x=<no value>, y=<no value>)

返回一个掩码数组，其中的元素来自x或者y，取决于条件。

注意

只有当健康)状况提供，该函数等同于numpy.nonzero。本文档的其余部分仅涵盖提供所有三个参数的情况。

参数：

condition： 数组，布尔

如果为 True，则产生 x，否则产生 y。

x, y： 数组，可选

可供选择的值。 x、y 和条件需要可广播到某种形状。

返回：

out： MaskedArray

一个掩码数组masked条件被屏蔽的元素，来自的元素x其中健康)状况是 True，并且元素来自y别处。

例子：

>>> x = np.ma.array(np.arange(9.).reshape(3, 3), mask=[[0, 1, 0],
...                                                    [1, 0, 1],
...                                                    [0, 1, 0]])
>>> x
masked_array(
  data=[[0.0, --, 2.0],
        [--, 4.0, --],
        [6.0, --, 8.0]],
  mask=[[False,  True, False],
        [ True, False,  True],
        [False,  True, False]],
  fill_value=1e+20)
>>> np.ma.where(x > 5, x, -3.1416)
masked_array(
  data=[[-3.1416, --, -3.1416],
        [--, -3.1416, --],
        [6.0, --, 8.0]],
  mask=[[False,  True, False],
        [ True, False,  True],
        [False,  True, False]],
  fill_value=1e+20)