使用二维数组进行布尔索引

Question

我有两个数组，a和b，一个2D，一个1D，包含两个相关量的值，它们以相同的顺序填充，因此a[0]与b[0]相关，依此类推

我想访问b的元素，其中a等于给定值，其中该值是1D数组本身。

例如

a=np.array([[0,0],[0,1],[1,0],[1,1]])
b=np.array([0, 7, 9, 4])
value = np.array([0,1])

在1D的情况下，我可以很容易地使用布尔索引并进行

b[a==value]

我想要的结果是7。

但在这种情况下，它不起作用，因为它在比较中检查b的每个元素，而不是检查子数组。。。

有快速的方法吗？

Answer 1

问题似乎与示例不匹配，但返回[7]:

b[(a == value).all(axis=-1)]