所以我需要邻接矩阵,这是一个二维数组,我想计算两个数组之间相似元素的数量。这听起来可能很傻,我知道可以用一个简单的for循环来完成,但我想知道是否有一个一行程序这样做?或者是一种更快的方法,因为我在处理大矩阵。以下代码是我现在拥有的:
adj1 = graph1.get_adjacency()
adj2 = graph2.get_adjacency()
count = 0
for i in range(len(adj1)):
for j in range(len(adj1)):
if adj[i][j] == adj[i][j]:
count += 1
最佳答案
试试这个:
np.sum(adj1 == adj2)
如果adj1和adj2的数据类型是float:
np.sum(np.isclose(adj1, adj2))