networkx - 查找深度为 2 Networkx 的所有子图

Question

我在networkx中有一个巨大的图，我想从每个节点获取深度为2的所有子图。有没有一种好的方法可以使用networkx中的内置函数来做到这一点？

Answer 1

正如我在评论中所说，networkx.ego_graph 符合要求。您只需确保将半径设置为 2(默认为 1):

import numpy as np
import matplotlib.pyplot as plt
import networkx as nx

# create some test graph
graph = nx.erdos_renyi_graph(1000, 0.005)

# create an ego-graph for some node
node = 0
ego_graph = nx.ego_graph(graph, node, radius=2)

# plot to check
nx.draw(ego_graph); plt.show()