利用pyecharts绘制桑基图---帮助小姐姐

pyecharts的中文网站，有这个包的详细介绍，需要什么直接去看什么。

绘制桑基图，主要是结点和联系。

原始数据的示意图：

pyecharts要求都要是json格式，这是一种数据格式，有专门的网站将excel转换成json。比如http://www.yzcopen.com/doc/exceljson
自己也找了个代码：

import codecs

file_name  = '.\\卵巢囊肿手术治疗（腹腔镜）.xlsx'
wb = xlrd.open_workbook(file_name)
dict_list = []
node_list = []
sh = wb.sheet_by_index(3)
title = sh.row_values(0)  #  获取表头，即第一行的字段
# print(title)  # ['科室', '项目类型', '金额']

# for循环遍历，从第1行开始(第0行是表头)
for i in range(1,sh.nrows):
    rowvalue = sh.row_values(i) #  每一行的数据
    single = OrderedDict()
    for j in range(len(rowvalue)):
        single[title[j]] = rowvalue[j] # 生成每一行的字典
    dict_list.append(single) # 将字典添加到列表里

print(dict_list)
j = json.dumps(dict_list,ensure_ascii=False)
# print(j)
with codecs.open('data2.json','w','utf-8') as f:
    f.write(j)

下图为生成的json数据格式：

生成桑基图：

import json

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

# 导入输出图片工具
from pyecharts.render import make_snapshot
# 使用snapshot-selenium 渲染图片
from snapshot_selenium import snapshot

with open("data.json", "r", encoding="utf-8") as f: #  加载数据
    j = json.load(f)
# print(j)
c = (
    Sankey()
    .add(
        "sankey",
        nodes = j["nodes"],  # 结点
        links = j["links"],  # 联系
        node_gap=10,
        linestyle_opt=opts.LineStyleOpts(opacity=0.4, curve=0.5, color="source"),
        label_opts=opts.LabelOpts(position="right"),
    )
    .set_global_opts(title_opts=opts.TitleOpts(title="Sankey-基本示例"),
                        )
    .render("gym.html")
)

make_snapshot(snapshot,file_name=c, output_name="gym.png")  # 保存图片

json

1.json.dumps将一个Python数据结构转换为JSON：

import json
data = {
    'name' : 'myname',
    'age' : 100,
}
json_str = json.dumps(data)

2.json.loads将一个JSON编码的字符串转换回一个Python数据结构：

data = json.loads(json_str)

3.json.dump() 和 json.load() 来编码和解码JSON数据,用于处理文件。

with open('test.json', 'w') as f:
    json.dump(data, f)
with open('test.json', 'r') as f:
    data = json.load(f)