Python | numpy.putmask () method

NumPy | Python Methods and Functions

With the numpy.putmask() method we can change the elements in the array by condition and set value with numpy.putmask () .

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 with numpy .putmask () we can get a new array using the given condition and value.

# NumPy import

import numpy as np

 
# using the 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:

# NumPy import

import numpy as np

 
# using the numpy.putmask () method

arr = np.array ([[ 1 , 2 , 3 ],

[ 3 , 2 , 1 ],

[ 1 , 2  , 3 ]])

 

np.putmask (arr, arr & gt; 2 , 4 )

 

print (arr)

Output:

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





Get Solution for free from DataCamp guru