如何在NumPy中按指定数组获取匹配元素的行索引？

你遇到的问题是因为直接用d == np.array([2,2])会做逐元素匹配，而不是判断整行是否完全吻合。我们需要先把逐元素的布尔结果合并成整行的匹配判断，再用np.argwhere提取索引。

解决步骤： #

• 判断整行是否完全匹配：用np.all()沿着行的方向（axis=1）对逐元素比较的结果做逻辑与操作，得到一个一维布尔数组，每个元素代表对应行是否完全等于目标数组[2,2]。
• 提取匹配行的索引：把这个布尔数组传入np.argwhere()，它会返回所有True位置的坐标，最后将结果扁平化得到一维索引数组。

完整代码示例： #

import numpy as np

d = np.array([[1,2],[2,2], [3,2], [2,2]])
# 第一步：逐元素比较后，判断整行是否完全匹配
row_matches = np.all(d == np.array([2,2]), axis=1)
# 第二步：提取匹配行的索引并扁平化
result = np.argwhere(row_matches).flatten()

print(result)
# 输出：[1 3]

为什么之前的方法不对？ #

你之前执行np.argwhere(d == np.array([2,2]))时，得到的是所有单个元素匹配的位置（比如第一行的第二个元素2也会被匹配到），而不是整行都匹配的行索引。通过np.all(axis=1)我们把行内的所有匹配结果合并，只保留整行完全符合的行标记，这样np.argwhere就能准确返回目标行的索引了。

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