我有兩個陣列,a一個b2D 和一個 1D,包含兩個相關數量的值,它們以相同的順序填充,例如與等a[0]相關b[0]。
我想訪問bwherea等于給定值的元素,其中值是一維陣列本身。
例如
a=np.array([[0,0],[0,1],[1,0],[1,1]])
b=np.array([0, 7, 9, 4])
value = np.array([0,1])
在一維情況下,我可以輕松使用布爾索引并執行
b[a==value]
我想要的結果是7。
但在這種情況下,它不起作用,因為它檢查b比較中的每個元素,而不是檢查子陣列......
有沒有快速的方法來做到這一點?
uj5u.com熱心網友回復:
這個問題似乎與示例不匹配,但這會回傳[7]:
b[(a == value).all(axis=-1)]
轉載請註明出處,本文鏈接:https://www.uj5u.com/net/521117.html
標籤:麻木的numpy索引
下一篇:將字串轉換為浮點值
