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

时间:2017-02-03 05:54:08

标签: python igraph

我正在使用AI,我的一个算法需要添加和删除边缘。我可以用igraph来做,但我的问题是:它太慢了。

如果我使用两个词典,我可以在0.2秒内运行我的代码。但是对于igraph,我需要超过5秒。而且我不知道如何提高代码性能。

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

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

感谢。

1 个答案:

答案 0 :(得分:2)

igraph的数据结构针对快速查询进行了优化,但不是那么快速的更新。无论何时添加顶点,删除顶点,添加边或删除边,igraph图中都有多个数据结构必须更新/重新编制索引。在许多情况下,删除单个边缘几乎与在一次通过中删除多个一样昂贵。

例如,如果我理解正确,您上面的目标是删除位于两个顶点组(self.vertexSIself.vertexSJ之间的所有边。你可以像上面那样做,但由于你逐个删除边缘,这将非常慢。您可以通过收集边缘的ID来加速它,然后先将其删除到列表中,然后在结尾处使用该列表调用self.g.delete_edges()。但是有一个更简单的单行(假设你的图是无向的):

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