numpy.ma.where#

ma.where(condition, x=<no value>, y=<no value>)[source]#

根据条件返回一个包含来自 x 或 y 的元素的掩码数组。

注意

当仅提供 condition 时，此函数与 nonzero 相同。本文档的其余部分仅涵盖所有三个参数都提供的案例。

参数：

conditionarray_like，bool: 为 True 时，生成 x，否则生成 y。
x, yarray_like，可选: 从中选择的数值。x、y 和 condition 需要广播到某个形状。

返回：

outMaskedArray: 一个掩码数组，其中条件为掩码时元素为 masked，condition 为 True 时元素来自 x，其他情况元素来自 y。

另请参阅

numpy.where: 顶级 NumPy 模块中的等效函数。
nonzero: 省略 x 和 y 时调用的函数

示例

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