Python中将数组中的某个值用其他值替换

问题

将数据中值为30的元素都替换成10

实现代码

import numpy as np

a = np.array([[20, 30, 40], [40, 50, 30], [20, 30, 40]])
print(a)
a[np.where(a == 30)] = 10
print(a)

结果： 

[[20 30 40]
 [40 50 30]
 [20 30 40]]
[[20 10 40]
 [40 50 10]
 [20 10 40]]