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)