pyecharts/test/test_graph.py
陈键冬 3fc897ae5b Lias (#722)
* Add: 新增 lias 工具格式化 imports

* Update: 细节修正

* Fix: fix #721

* Update: var reused

* Format: 格式化导包顺序

* Update: modules import order: python builtin imports on top, 3rd parties libraries in the middle, package's own libraries in the bottom
2018-09-01 16:07:00 +08:00

70 lines
1.7 KiB
Python

#!/usr/bin/env python
# coding=utf-8
from __future__ import unicode_literals
import codecs
import json
import os
from pyecharts import Graph
nodes = [
{"name": "结点1", "symbolSize": 10},
{"name": "结点2", "symbolSize": 20},
{"name": "结点3", "symbolSize": 30},
{"name": "结点4", "symbolSize": 40},
{"name": "结点5", "symbolSize": 50},
{"name": "结点6", "symbolSize": 40},
{"name": "结点7", "symbolSize": 30},
{"name": "结点8", "symbolSize": 20},
]
def test_graph_force_layout():
links = []
for i in nodes:
for j in nodes:
links.append({"source": i.get("name"), "target": j.get("name")})
graph = Graph("关系图-力引导布局示例")
graph.add("", nodes, links, repulsion=8000, line_color="#aaa")
graph.render()
def test_graph_circular_layout():
links = []
for i in nodes:
for j in nodes:
links.append({"source": i.get("name"), "target": j.get("name")})
graph = Graph("关系图-环形布局示例")
graph.add(
"",
nodes,
links,
is_label_show=True,
graph_repulsion=8000,
graph_layout="circular",
label_text_color=None,
)
graph.render()
def test_graph_official_data():
with codecs.open(
os.path.join("fixtures", "weibo.json"), "r", encoding="utf-8"
) as f:
j = json.load(f)
nodes, links, categories, cont, mid, _ = j
graph = Graph("微博转发关系图", width=1200, height=600)
graph.add(
"",
nodes,
links,
categories,
label_pos="right",
graph_repulsion=50,
is_legend_show=False,
line_curve=0.2,
label_text_color=None,
)
graph.render()