所以我需要邻接矩阵,这是一个二维数组,我想计算两个数组之间相似元素的数量。这听起来可能很傻,我知道可以用一个简单的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))

08-08 03:13