-
Notifications
You must be signed in to change notification settings - Fork 0
/
charts.py
92 lines (87 loc) · 2.67 KB
/
charts.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# -*- coding: utf-8 -*-
"""
Created on Thu Oct 20 10:23:44 2022
@author: shangfr
"""
from streamlit_echarts import st_echarts
def render_graph(graph):
option = {
"backgroundColor": graph['bg_color'],
"title": {
"text": "",
"subtext": "",
"top": "top",
"left": "right",
},
"toolbox": {
"show": True,
"feature": {
"dataView": {
"show": True,
},
"restore": {
"show": True
},
"saveAsImage": {
"show": False
}
}
},
"tooltip": {},
"legend": [{"data": [a["name"] for a in graph["categories"]]}],
"series": [
{
"name": "Packages",
"type": "graph",
"layout": graph["layout"],
"data": graph["nodes"],
"links": graph["links"],
"categories": graph["categories"],
"lineStyle": {"color": graph["links_color"]},
"roam": True,
# "label": {"position": "right"},
"label": {"show": graph["show_n"], "fontSize": graph["nodes_font_size"]},
"edgeSymbol": ["none", "arrow"],
"draggable": True,
"force": {"repulsion": graph["repulsion_forces"]},
}
],
}
st_echarts(option, height="600px")
def render_tree(graph):
data = graph["data"]
#rename_key_recursive(data[0])
option = {
"backgroundColor": graph['bg_color'],
"tooltip": {"trigger": "item", "triggerOn": "mousemove"},
"series": [
{
"type": "tree",
"data": data,
"top": "1%",
"left": "7%",
"bottom": "1%",
"right": "20%",
"symbolSize": 7,
"layout": graph["layout"],
"label": {
"position": "left",
"verticalAlign": "middle",
"align": "right",
"fontSize": 9,
},
"leaves": {
"label": {
"position": "right",
"verticalAlign": "middle",
"align": "left",
}
},
"emphasis": {"focus": "descendant"},
"expandAndCollapse": True,
"animationDuration": 550,
"animationDurationUpdate": 750,
}
],
}
st_echarts(option, height="800px")