pyecharts/test/test_treemap.py
2017-10-21 19:30:06 +08:00

71 lines
1.9 KiB
Python

#!/usr/bin/env python
# coding=utf-8
from __future__ import unicode_literals
import sys
from pyecharts import TreeMap
PY2 = sys.version_info[0] == 2
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 test_treemap_default():
treemap = TreeMap("树图-默认示例", width=1200, height=600)
treemap.add("演示数据", data, is_label_show=True, label_pos='inside')
treemap.render()
def test_treemap_drilldown():
treemap = TreeMap("树图-下钻示例", width=1200, height=600)
treemap.add("演示数据", data, is_label_show=True, label_pos='inside',
treemap_left_depth=1)
treemap.render()
def test_treemap_offcical_data():
treemap = TreeMap("树图-官方数据", width=1200, height=600)
import os
import json
if PY2:
import codecs
with codecs.open(os.path.join("..", "json", "treemap.json"), "rb") as f:
data = json.load(f)
else:
with open(os.path.join("..", "json", "treemap.json"),
"r", encoding="utf-8") as f:
data = json.load(f)
treemap.add("演示数据", data, is_label_show=True, label_pos='inside')
treemap.render()