numpy学习之 np.where()

numpy学习之 np.where()

如何再不影响原始数组的情况下改变元素项

用到的时np中的 where() 数，用法如下：

np.where(condition, x, y)
满足条件(condition)，输出x，不满足输出y。

arr = np.array([0,1,2,3,4,5,6])
out = np.where(arr % 2 == 1, -1 , arr) ##条件是数组中奇数，
                            #将其值设置为 -1.不是的设置为 数组中的原来的值。

输出结果是：

out:
array([ 0, -1,  2, -1,  4, -1,  6])
array:
array([0, 1, 2, 3, 4, 5, 6])