mirror of
https://github.com/pyecharts/pyecharts.git
synced 2025-12-08 20:59:23 +00:00
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
# coding=utf-8
|
|
from ... import options as opts
|
|
from ...charts.chart import Chart
|
|
from ...commons.types import Numeric, Optional, Sequence, Union
|
|
from ...globals import ChartType
|
|
|
|
|
|
class Tree(Chart):
|
|
"""
|
|
<<< 树图 >>>
|
|
|
|
树图主要用来可视化树形数据结构，是一种特殊的层次类型，具有唯一的根节点，左子树，和右子树。
|
|
"""
|
|
|
|
def __init__(self, init_opts: Union[opts.InitOpts, dict] = opts.InitOpts()):
|
|
super().__init__(init_opts=init_opts)
|
|
|
|
@staticmethod
|
|
def _set_collapse_interval(data, interval):
|
|
"""
|
|
间隔折叠节点，当节点过多时可以解决节点显示过杂间隔。
|
|
|
|
:param data: 节点数据
|
|
:param interval: 指定间隔
|
|
"""
|
|
if interval <= 0:
|
|
return data
|
|
if data and isinstance(data, list):
|
|
for d in data:
|
|
children = d.get("children", None)
|
|
if children and interval > 0:
|
|
for index, value in enumerate(children):
|
|
if index % interval == 0:
|
|
value.update(collapsed="false")
|
|
return data
|
|
|
|
def add(
|
|
self,
|
|
series_name: str,
|
|
data: Sequence,
|
|
*,
|
|
layout: str = "orthogonal",
|
|
symbol: str = "emptyCircle",
|
|
symbol_size: Numeric = 7,
|
|
orient: str = "LR",
|
|
pos_top: Optional[str] = None,
|
|
pos_left: Optional[str] = None,
|
|
pos_bottom: Optional[str] = None,
|
|
pos_right: Optional[str] = None,
|
|
collapse_interval: Numeric = 0,
|
|
label_opts: Union[opts.LabelOpts, dict] = opts.LabelOpts(),
|
|
leaves_label_opts: Union[opts.LabelOpts, dict] = opts.LabelOpts(),
|
|
tooltip_opts: Union[opts.TooltipOpts, dict, None] = None,
|
|
):
|
|
if isinstance(label_opts, opts.LabelOpts):
|
|
label_opts = label_opts.opts
|
|
if isinstance(leaves_label_opts, opts.LabelOpts):
|
|
leaves_label_opts = leaves_label_opts.opts
|
|
if isinstance(tooltip_opts, opts.TooltipOpts):
|
|
tooltip_opts = tooltip_opts.opts
|
|
|
|
_data = self._set_collapse_interval(data, collapse_interval)
|
|
self.options.get("series").append(
|
|
{
|
|
"type": ChartType.TREE,
|
|
"name": series_name,
|
|
"data": _data,
|
|
"left": pos_left,
|
|
"right": pos_right,
|
|
"top": pos_top,
|
|
"bottom": pos_bottom,
|
|
"symbol": symbol,
|
|
"symbolSize": symbol_size,
|
|
"layout": layout,
|
|
"orient": orient,
|
|
"label": label_opts,
|
|
"leaves": {"label": leaves_label_opts},
|
|
"tooltip": tooltip_opts,
|
|
}
|
|
)
|
|
return self
|