[[11. 5.]
[24. 6.]
[39. 12.]
[14. 1.]
[25. 12.]]
那是我的矩陣,現在我想對其應用以下條件: 10 < a < 15但僅適用于第一列
結果我想要一個索引為 [0. 3.]
我對“where”的嘗試失敗了,numpy 將它應用于兩列
uj5u.com熱心網友回復:
第一列是a[:, 0]。np.flatnonzero是np.nonzero(where只有一個引數)的一個版本,它回傳一個平面陣列而不是一個索引元組。
你可以做類似的事情
mask = (10 > a[:, 0]) & (a[:, 0] < 15)
idx = np.flatnonzero(mask)
另一種方式是
idx, = np.nonzero(mask)
甚至
idx, = np.where(mask)
請注意idx最后兩個示例中的逗號,它會觸發引數解包。
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/349957.html
