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])代表縱座標

更高維度可以依次從外往內看進去.