python - 将 networkx DiGraph 打印为嵌套的 JSON 格式

Question

我有一个 networkx 有向图。

>> G = nx.DiGraph()
>> G.add_edges_from(product([0],[1,2])) # product is itertools.product 
>> G.add_edges_from(product([1],[3,4])) 

里面的图是这样的:

>> G.adj  
 {0: {1: {}, 2: {}},  
  1: {3: {}, 4: {}},  
  2: {},  
  3: {},  
  4: {}} 

我希望将其打印成如下所示的 json 格式:

{   
    name: 0,
    children: [ 
        {
            name: 1,
            children: [
                {
                    name: 3,
                    children: [] 
                }, 
                {
                    name: 4,
                    children: []
                }
                ]
        }, 
        {
            name: 2,
            children:[]
        }
    ]
} 

(原因是，我试图以这种 D3 可视化采用的格式编写它。https://gist.github.com/mbostock/4063550)

我不知道该怎么做。非常感谢任何帮助。

谢谢。

Answer 1

您可以使用 json_graph.tree_data() 以嵌套格式输出数据。

来自 the documentation

from networkx.readwrite import json_graph
import networkx as nx
import json
G = nx.DiGraph([(1,2)])
data = json_graph.tree_data(G,root=1)
s = json.dumps(data)

它给出 {"id": 1, "children": [{"id": 2}]}