PyEcharts 树形图表之矩形树图——头歌

第1关：TreeMap：矩形树图（一）

from PreTest import *
from pyecharts import options as opts
from pyecharts.render import make_snapshot
from snapshot_phantomjs import snapshot

from pyecharts.charts import TreeMap


data = [
    {"value": 40, "name": "我是A"},
    {
        "value": 180,
        "name": "我是B",
        "children": [
            {
                "value": 76,
                "name": "我是B.children",
                "children": [
                    {"value": 12, "name": "我是B.children.a"},
                    {"value": 28, "name": "我是B.children.b"},
                    {"value": 20, "name": "我是B.children.c"},
                    {"value": 16, "name": "我是B.children.d"},
                ],
            }
        ],
    },
]

def treemap_chart() -> TreeMap:
    # ********* Begin *********#  
    treemap = (
        TreeMap()
        .add(
            series_name="演示数据",
            data=data,
            label_opts=opts.LabelOpts(formatter="{b}", position="inside"),
        )
        .set_global_opts(title_opts=opts.TitleOpts(title="TreeMap-基本示例"))
    )
    # ********** End **********#
    return treemap

make_snapshot(snapshot, treemap_chart().render("Result/render.html"), 'StudentAnswer/student_answer.png') # 输出图片
make_snapshot(snapshot, treemap_base(data).render(), "StandardAnswer/task1/standard_answer_1.png")

第2关：TreeMap：矩形树图（二）

from PreTest import *
from pyecharts import options as opts
from pyecharts.render import make_snapshot
from snapshot_phantomjs import snapshot

from pyecharts.charts import TreeMap
import json


with open("treemap.json", "r", encoding="utf-8") as f:
    data = json.load(f)


def treemap_chart() -> TreeMap:
    # ********* Begin *********#  
    treemap = (
        TreeMap()
        .add(
            series_name="演示数据",
            data=data,
            levels=[
                # 第一层配置
                opts.TreeMapLevelsOpts(
                    color_saturation=[0.3, 0.6],
                    treemap_itemstyle_opts=opts.TreeMapItemStyleOpts(
                        border_color="#555",
                        border_width=4,
                        gap_width=4
                    ),
                    label_opts=opts.LabelOpts(is_show=True, position="inside")
                ),
                # 第二层配置
                opts.TreeMapLevelsOpts(
                    color_saturation=[0.3, 0.6],
                    treemap_itemstyle_opts=opts.TreeMapItemStyleOpts(
                        border_color_saturation=0.7,
                        border_width=2,
                        gap_width=2
                    ),
                    label_opts=opts.LabelOpts(is_show=True, position="inside")
                ),
                # 第三层配置
                opts.TreeMapLevelsOpts(
                    color_saturation=[0.3, 0.5],
                    treemap_itemstyle_opts=opts.TreeMapItemStyleOpts(
                        border_color_saturation=0.6,
                        gap_width=1
                    ),
                    label_opts=opts.LabelOpts(is_show=True, position="inside")
                ),
                # 第四层配置
                opts.TreeMapLevelsOpts(
                    color_saturation=[0.3, 0.5],
                    label_opts=opts.LabelOpts(is_show=True, position="inside")
                )
            ]
        )
        .set_global_opts(title_opts=opts.TitleOpts(title="TreeMap-Levels-配置"))
    )
    # ********** End **********#
    return treemap

make_snapshot(snapshot, treemap_chart().render("Result/render.html"), "StudentAnswer/student_answer.png") # 输出图片
make_snapshot(snapshot, treemap_levels(data).render(), "StandardAnswer/task2/standard_answer_2.png")