Python numpy place用法及代碼示例

本文簡要介紹 python 語言中 numpy.place 的用法。

用法:

numpy.place(arr, mask, vals)

根據條件值和輸入值更改數組元素。

相似np.copyto(arr, vals, where=mask), 區別在於place使用前 N 個元素瓦爾斯, 其中 N 是真值的數量掩碼， 盡管copyto使用元素掩碼為真。

請注意， extract 的作用與 place 完全相反。

參數：

arr： ndarray

將數據放入的數組。

mask： array_like

布爾掩碼數組。必須具有與 a 相同的大小。

vals： 一維序列

要放入 a 中的值。僅使用前 N 個元素，其中 N 是掩碼中 True 值的數量。如果 vals 小於 N，它將被重複，如果要屏蔽 a 的元素，則該序列必須是非空的。

例子：

>>> arr = np.arange(6).reshape(2, 3)
>>> np.place(arr, arr>2, [44, 55])
>>> arr
array([[ 0,  1,  2],
       [44, 55, 44]])