我正在使用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
感谢。
答案 0 :(得分:2)
igraph的数据结构针对快速查询进行了优化,但不是那么快速的更新。无论何时添加顶点,删除顶点,添加边或删除边,igraph图中都有多个数据结构必须更新/重新编制索引。在许多情况下,删除单个边缘几乎与在一次通过中删除多个一样昂贵。
例如,如果我理解正确,您上面的目标是删除位于两个顶点组(self.vertexSI
和self.vertexSJ
之间的所有边。你可以像上面那样做,但由于你逐个删除边缘,这将非常慢。您可以通过收集边缘的ID来加速它,然后先将其删除到列表中,然后在结尾处使用该列表调用self.g.delete_edges()
。但是有一个更简单的单行(假设你的图是无向的):
self.g.es.select(_between=(self.vertexSI, self.vertexSJ)).delete()