通过顶点索引更快地删除边的方法[iggraph Python]

2022-04-16 00:00:00 python igraph

问题描述

我正在使用人工智能,我的一个算法需要添加和删除边。我可以用数字来做,但我的问题是:它太慢了。

如果我使用两个词典,我可以在0.2秒内运行代码。但用字形,我用的不止是5s。我不知道如何提高代码性能。

以下代码是算法的一部分。这将移除两个顶点列表之间的所有边。有谁知道如何以更好的性能做到这一点?

for vertexI in self.vertexSI:
    for vertexJ in self.vertexSJ:
        try:
            nOfLoops += 1
            edgeID = self.g.get_eid(vertexI.index, vertexJ.index)
            self.g.delete_edges(edgeID)
        except Exception as e:
            nOfErrors += 1

谢谢。


解决方案

iggraph的数据结构针对快速查询进行了优化,但更新速度不是很快。当您添加顶点、删除顶点、添加边或删除边时,曲线图中有几个数据结构必须更新/重新索引。在许多情况下,删除单个边几乎与一次删除多个边的成本一样高。

例如,如果我理解正确的话,您上面的目标是删除落在两个顶点组(self.vertexSIself.vertexSJ)之间的所有边。你可以像上面那样做,但这将是非常慢的,因为你是一个接一个地删除边。您可以通过首先将要删除的边的ID收集到一个列表中,然后在该列表的末尾调用self.g.delete_edges()来加快速度。但还有一个更简单的一行程序(假设您的图是无向的):

self.g.es.select(_between=(self.vertexSI, self.vertexSJ)).delete()

相关文章