1. 程式人生 > >np.where()用法總結

np.where()用法總結

用法一:可以返回一個n維陣列,可廣播。

# np.where(condition, x, y)
# condition, x, y, return are all can be #ndarray
x = np.random.randn((4, 4))
np.where(x>0, 2, -2)

output:


用法二:condition 也可以是布林型陣列,每個條件都和x,y對應(廣播)


用法三:找到n維陣列中特定數值的索引