如何通过匹配二维数组子数组实现布尔索引获取对应一维数组元素？

要匹配二维数组a中的整行子数组，你可以用numpy的逐元素比较结合行维度的all()方法生成布尔索引，再用这个索引提取b中对应的元素。

核心实现代码 #

import numpy as np

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

# 生成每行完全匹配value的布尔掩码
mask = (a == value).all(axis=1)
# 提取对应b的元素
target = b[mask]
print(target)  # 输出：[7]

原理说明 #

• a == value会对a和value进行逐元素比较，得到一个和a同形状的布尔数组（每个元素表示对应位置是否相等）。
• .all(axis=1)会沿着行的方向（axis=1）检查每一行的所有元素是否都为True，也就是整行子数组和value完全匹配，最终得到一个一维布尔数组，长度和b一致。
• 用这个一维布尔数组作为索引，就能精准提取b中对应a匹配行的元素。

浮点数场景适配 #

如果你的数组是浮点数，直接用==可能会有精度问题，推荐用np.allclose()来生成掩码：

mask = np.allclose(a, value, atol=1e-8)  # atol是允许的绝对误差
target = b[mask]

内容的提问来源于stack exchange，提问作者etien