np.where(condition, [x, y])官方解释看的不是很懂，研究各种材料后，总结如下：
一、np.where(condition, x, y)
1.返回值是一个和condition的shape相同的numpy 数组
2.当满足条件condition时，返回值中的元素从x中取，否则从y中取
二、np.where(condition)
1. 当只有condition时，返回值是满足condition的元素的下标
2. 返回值类型是一个tuple，例如a是2-D，则tuple的第一个元素代表行下标，第二个元素代表列下标
