Python中怎么从excel导入数据,使之成为有向图的权重
时间: 2024-05-01 17:16:13 浏览: 81
Python向excel中写入数据的方法
可以使用Python中的pandas库和networkx库来完成操作。首先使用pandas库读取excel数据,然后使用networkx库创建有向图,并根据数据给图中的边赋权,最后可以使用networkx库进行可视化展示。
以下是示例代码:
```python
import pandas as pd
import networkx as nx
# 读取excel数据
data = pd.read_excel('data.xlsx', index_col=0)
# 创建有向图
G = nx.DiGraph()
# 给图中的边赋权
for i, row in data.iterrows():
node_from = row['node_from']
node_to = row['node_to']
weight = row['weight']
G.add_edge(node_from, node_to, weight=weight)
# 可视化展示
nx.draw(G, with_labels=True)
```
其中,`data.xlsx`是包含有向图数据的excel文件,其格式应该为:
| node_from | node_to | weight |
| --------- | ------- | ------ |
| A | B | 1 |
| B | C | 2 |
| C | D | 3 |
| ... | ... | ... |
其中,`node_from`和`node_to`分别表示边的起点和终点,`weight`表示边的权重。
阅读全文