np.where()详解

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的第一个元素代表行下标，第二个元素代表列下标