Python| numpy.putmask() 方法

在numpy.putmask()方法的帮助下，我们可以使用numpy.putmask()方法在条件和给定值的帮助下更改数组中的元素。

Syntax : numpy.putmask(array, condition, value)
Return : 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]]