np.where(condition, [x, y])官方解释看的不是很懂，研究各种材料后，总结如下：

一、np.where(condition, x, y)

1.返回值是一个和condition的shape相同的numpy 数组

2.当满足条件condition时，返回值中的元素从x中取，否则从y中取

 np.where()详解-风君雪科技博客

二、np.where(condition)

1. 当只有condition时，返回值是满足condition的元素的下标

2. 返回值类型是一个tuple，例如a是2-D，则tuple的第一个元素代表行下标，第二个元素代表列下标

np.where()详解-风君雪科技博客

np.where()详解-风君雪科技博客