在python ndarray中查找重复行的索引

import numpy as np

# 创建一个ndarray数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [1, 2, 3],
                [7, 8, 9],
                [4, 5, 6]])

# 使用numpy.unique()函数找到唯一行和索引
unique_rows, unique_indices = np.unique(arr, axis=0, return_index=True)

# 使用numpy.where()函数找到重复行的索引
duplicate_indices = np.where(np.in1d(np.arange(len(arr)), unique_indices, invert=True))[0]

# 打印结果
print("重复行的索引：", duplicate_indices)

重复行的索引： [2 4]