np.where()

官网解释:

使用方式:

1) np.where(condition, x, y):  满足condition输出 x ,否则输出 y.

例子:

      

 

1) np.where(arry): 返回arry中真值的索引或者座标

例子:

注意:对于第一个print的输出. array([0, 1, 1])代表横座标, array([0, 0, 1])代表纵座标

同理,对于第二个print的输出. array([0, 1, 1])代表横座标, array([0, 0, 1])代表纵座标

更高维度可以依次从外往内看进去.