numpy 寻找两个二维数组中重复的行（附代码)

numpy 两个 N*M 维度的 二维数组a 和 b， 以行为单位，找到 a 和 b 中都存在的行，相同的行不一定出现在同一行的位置，通过遍历每一行寻找。

import numpy as np

# 创建两个示例数组
a_sorted = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 6], [10, 11, 11]])
b_sorted = np.array([[10, 11, 11], [7, 8, 6], [1, 2, 3], [13, 14, 15]])

# 初始化一个空数组，用于存储相同的行
common_rows = []

# 遍历a_sorted中的每一行
for row in a_sorted:
    # 使用 np.all() 和 np.any() 函数找到相同的行
    same_rows = np.all(b_sorted == row, axis=1)
    # print(same_rows)
    if np.any(same_rows):
        common_rows.append(row)

print("Common rows:")
print(np.vstack(common_rows))