python 矩阵比较相同行或列是否相同

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(),进行比较。

发布了23 篇原创文章 · 获赞 9 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/Darren1921/article/details/88417106