利用python画一个桑基图,给一个案例
时间: 2023-04-08 17:01:39 浏览: 232
python 画图,综合例子(示例)
好的,以下是一个利用Python绘制桑基图的案例:
```python
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.sankey import Sankey
# 创建数据
data = {'A': [10, 20, 30], 'B': [20, 30, 10], 'C': [30, 10, 20]}
df = pd.DataFrame(data, index=['Year 1', 'Year 2', 'Year 3'])
# 创建桑基图
fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(1, 1, 1, xticks=[], yticks=[], title="Sankey Diagram")
sankey = Sankey(ax=ax, scale=0.01, offset=0.2, head_angle=180)
sankey.add(flows=df.iloc[0], labels=df.columns.tolist(), orientations=[-1, 1, 0])
sankey.add(flows=df.iloc[1], orientations=[-1, 1, 0])
sankey.add(flows=df.iloc[2], orientations=[-1, 1, 0])
diagrams = sankey.finish()
# 显示图像
plt.show()
```
这个案例中,我们使用了Pandas和Matplotlib库来创建数据和绘制桑基图。我们创建了一个包含三个类别(A、B、C)和三个时间点(Year 1、Year 2、Year 3)的数据集,然后使用Sankey函数来绘制桑基图。最后,我们使用Matplotlib的show函数来显示图像。
希望这个案例能够帮助你了解如何使用Python绘制桑基图。
阅读全文