np.where函数是三元表达式x if condition else y的向量化版本。np.where的第二,三个参数并不需要是数组,也可以是标量。
1、假设随机生成一个矩阵数据,想将其中所有正值替换为2,其中所有负值替换为-2。
其中,np.where(arr>0,2,-2)可以理解为 2 if arr>0 else -2
2、标量和数组联合。如下图,将数组中所有正值替换成2。
np.where函数是三元表达式x if condition else y的向量化版本。np.where的第二,三个参数并不需要是数组,也可以是标量。
1、假设随机生成一个矩阵数据,想将其中所有正值替换为2,其中所有负值替换为-2。
其中,np.where(arr>0,2,-2)可以理解为 2 if arr>0 else -2
2、标量和数组联合。如下图,将数组中所有正值替换成2。