问题描述
我知道 networkx 可以提供基于深度控制的广度优先搜索 (bfs) 结果。我想知道是否有解决方法,以便我可以通过边数控制结果?例如,我希望通过 bfs 获得一个节点 i 周围的 10 条边。但我不知道它的深度。
深度控制的bfs有点像
bfs = nx.bfs_edges(G,source=i,depth_limit=5)
我希望使用类似的功能
bfs = nx.bfs_edges(G,number=k)
因为我希望找到一个节点周围的所有边。所以看起来 nx.edge_bfs
是更好的选择?此函数返回当前的所有边。我们可以以某种方式修改它吗?我希望源节点尽可能位于中心,即yield边缘可以均匀地围绕源节点。
解决方法
查看nx.edge_bfs
的源代码,看起来这个函数本身返回了一个迭代器,它产生了源节点周围的边。它基本上像圆形一样返回周围的边缘。
如果有人能证实这一点会更好。这是源link