import numpy as np
a = np.array([[0,1,2],[1,2,3],[2,3,4]]).T
b = np.array([[4,5,6],[0,1,2],[1,2,3],[3,4,5],[2,3,4]])
index = []
for i in np.arange(len(b)):
for j in np.arange(len(a.T)):
if (b[i]==a[j]).all():
index.append(i)
print(index)
比较两个矩阵的行时,使用(b[i]==a[j]).all()或者(b[i]==a[j]).any(),进行比较。