Python numpy.putmask()方法
在numpy.putmask()方法的帮助下,我们可以通过numpy.putmask()方法在条件和给定值的帮助下改变数组中的元素。
语法: numpy.putmask(array, condition, 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]]