1. 程式人生 > >numpy.where

numpy.where

numpy.where有兩種用法,

1. np.where(condition, x, y)

如果condition是一維陣列,相當於[xv if c else yv for (c,xv,yv) in zip(condition,x,y)]

如果condition是高維陣列,同樣是把x和y做zip操作,然後根據條件的T/F進行選擇

 

2.np.where(condition)

只有條件 (condition),則輸出滿足條件 (即非0) 元素的座標 (等價於numpy.nonzero)。這裡的座標以tuple的形式給出,通常原陣列有多少維,輸出的tuple中就包含幾個陣列,每個陣列的第i個位置對應符合條件元素的各維座標。

 


參考連結:

https://www.cnblogs.com/massquantity/p/8908859.html