networkx.DiGraph保存为文件
时间: 2024-03-27 14:41:18 浏览: 231
可以使用 `networkx` 库提供的 `write_adjlist` 或 `write_edgelist` 方法将 `networkx.DiGraph` 对象保存为文件。这两种方法可以将有向图存储为文本文件,以便稍后读取和处理。
以下是将 `networkx.DiGraph` 对象保存为邻接表格式的示例代码:
```python
import networkx as nx
# 创建有向图
G = nx.DiGraph()
G.add_edges_from([(1, 2), (2, 3), (3, 1)])
# 将有向图保存为邻接表格式的文本文件
nx.write_adjlist(G, "directed_graph.adjlist")
```
上述代码中,我们使用 `networkx` 库创建了一个有向图,并将其保存为邻接表格式的文本文件。`write_adjlist` 方法将有向图写入一个文本文件中,每一行表示一个节点及其邻居节点。例如,对于有向图中的节点 `1`,其邻居节点为 `2` 和 `3`,则文本文件中的一行为:`1 2 3`。文本文件中的每行以换行符 `\n` 结尾。
如果要将有向图保存为边列表格式的文本文件,则可以使用 `write_edgelist` 方法。以下是示例代码:
```python
import networkx as nx
# 创建有向图
G = nx.DiGraph()
G.add_edges_from([(1, 2), (2, 3), (3, 1)])
# 将有向图保存为边列表格式的文本文件
nx.write_edgelist(G, "directed_graph.edgelist")
```
在上述代码中,我们使用 `write_edgelist` 方法将有向图保存为边列表格式的文本文件。每一行表示一条边,以源节点、目标节点和可选的权重值组成,用空格分隔。例如,对于有向图中的一条边 `(1, 2)`,其在文本文件中的表示为:`1 2`。
阅读全文