#numpy入门4.3.1p111-np.where函数

numpy.where函数是三元表达式 x if condition else y 的向量化版本。
当数组是多维时,依然可以奏效。
np.where(cond, xarr, yarr)
np.where的第二个和第三个参数并不需要是数组,它们可以是标量。where在数据分析中的一个典型用法是根据一个数组来生成一个新的数组。
假设你有一个随机生成的矩阵数据,并且你想将其中的正值都替换为2,将所有的负值替换为-2:

arr = np.random.randn(4,4)
np.where(arr > 0, 2,-2)    #arr>0生成一个布尔值矩阵,大于0的为True,小于0的为False

猜你喜欢

转载自blog.csdn.net/Poisson_SHAN/article/details/82813190