NetworKit 迭代器接受一个回调函数,所以如果你想迭代边缘(或节点),你必须定义一个函数,然后将它作为参数传递给迭代器。您可以在此处找到更多信息。例如,一个只打印所有边的简单函数是:
# Callback function.
# To iterate over edges it must accept 4 parameters
def myFunction(u, v, weight, edgeId):
print("Edge from {} to {} has weight {} and id {}".format(u, v, weight, edgeId))
# Using iterator with callback function
G.forEdges(myFunction)
现在,如果您想继续删除权重在权重列表内的边,直到图形分成两个连通分量,您还必须更新图形的连通分量,因为 ConnectedComponents 不会自动为您执行此操作(这可能也是其中之一迭代需要永远的原因)。要有效地执行此操作,您可以使用 DynConnectedComponents 类(请参见下面的示例)。在这种情况下,我认为边缘迭代器对你没有多大帮助,所以我建议你继续使用 for 循环。
from networkit import *
# Efficiently updates connected components after edge updates
cc = components.DynConnectedComponents(G).run()
# Removes edges with weight equals to w until components split
def removeEdges(w):
for (u, v) in G.edges():
if G.weight(u, v) == weight:
G.removeEdge(u, v)
# Updating connected components
event = dynamic.GraphEvent(dynamic.GraphEvent.EDGE_REMOVAL, u, v, weight)
cc.update(event)
if cc.numberOfComponents() > 1:
# Components did split
return True
# Components did not split
return False
if cc.numberOfComponents() == 1:
for weight in weights:
if removeEdges(weight):
break
这应该会加快您的原始代码速度。但是,它仍然是顺序代码,因此即使您在多核机器上运行它,它也只会使用一个内核。