np.where函数

np.where函数是numpy库中的一个非常实用的函数，主要有两个用法。第一个用法是np.where(condition, x, y)，它会返回满足condition的索引。第二个用法是首先传入y，然后如果满足条件condition，那么y中原来的数值变为x，否则不变化。相当于[xv if c else yv for (c,xv,yv) in zip(condition,x,y)]。这个函数可以帮助我们快速地对数组进行筛选和替换操作。