我有两个数组,a
和b
,一个2D和一个1D,包含以相同顺序填充的两个相关量的值,例如a[0]
与b[0]
相关,依此类推。
我想访问b
的元素,其中a
等于给定值,其中的值是一维数组本身。
例如
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
的每个元素,而不是检查子数组……
有什么快捷的方法可以做到这一点吗?
1条答案
按热度按时间ibps3vxo1#
问题似乎与示例不匹配,但这将返回
[7]
: