With the help of numpy.putmask()
method, we can change the elements in an array with the help of condition and given value by using numpy.putmask()
method.
Syntax :
numpy.putmask(array, condition, value)
Return : Return the array having new elements according to value.
Example #1 :
In this example we can see that by using numpy.putmask()
method, we are able to get the new array with the help of a given condition and value.
# 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)
|
Output :
array([1, 0, 3, 0, 5, 0])
Example #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)
|
Output :
[[1 2 4]
[4 2 1]
[1 2 4]]
Article Tags :