Open In App

Python | numpy.putmask() method

Last Updated : 03 Oct, 2019
Comments
Improve
Suggest changes
Like Article
Like
Report

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]]



Next Article
Article Tags :
Practice Tags :

Similar Reads