桑基图 python

桑基图是一种可视化图表，用于展示不同类别之间的关系和流动。在Python中，有多种工具可以用来制作桑基图，比如Excel、Tableau和Pyecharts等。Pyecharts是一个强大的Python库，用于创建各种交互式图表，包括桑基图。

要使用Pyecharts绘制桑基图，首先需要准备数据。可以从各种来源获取数据，比如Kaggle。然后，使用Pyecharts库中的Sankey模块来创建桑基图，并传入数据进行可视化。

下面是一个使用Pyecharts绘制桑基图的示例代码：

from pyecharts import options as opts
from pyecharts.charts import Sankey
import pandas as pd

# 读取数据
data = pd.read_excel('/Users/wangwangyuqing/Desktop/train.xlsx')

# 创建桑基图对象
sankey = (
    Sankey(init_opts=opts.InitOpts(width="1200px", height="800px", theme='westeros'))
    .add(
        "",
        nodes=nodes,
        links=links,
        linestyle_opt=opts.LineStyleOpts(opacity=0.2, curve=0.5, color="source"),
        label_opts=opts.LabelOpts(position="right"),
    )
    .set_global_opts(title_opts=opts.TitleOpts(title="桑基图"))
)

# 生成图表
sankey.render("/Users/wangwangyuqing/Desktop/image.html")

通过以上代码，你可以根据你的数据创建一个桑基图，并将其保存为一个HTML文件。在这个图表中，你可以展示不同类别之间的关系和流动。