Python numpy.putmask()用法及代碼示例

借助numpy.putmask()方法，我們可以在條件和給定值的幫助下，通過使用來更改數組中的元素numpy.putmask()方法。

用法： numpy.putmask(array, condition, value)

返回：Return the array having new elements according to value.

範例1：
在這個例子中，我們可以通過使用numpy.putmask()方法，我們可以在給定條件和值的幫助下獲取新數組。

# import numpy 
import numpy as np 
  
# using numpy.putmask() method 
arr = np.array([1, 2, 3, 4, 5, 6]) 
np.putmask(arr, arr % 2 == 0, 0) 
  
print(arr)

輸出：

array([1, 0, 3, 0, 5, 0])

範例2：

# import numpy 
import numpy as np 
  
# using numpy.putmask() method 
arr = np.array([[1, 2, 3], 
                [3, 2, 1], 
                [1, 2, 3]]) 
  
np.putmask(arr, arr>2, 4) 
  
print(arr)

輸出：

[[1 2 4]
[4 2 1]
[1 2 4]]

相關用法

注：本文由純淨天空篩選整理自Jitender_1998大神的英文原創作品 Python | numpy.putmask() method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。