from pyecharts import options as opts from pyecharts.charts import Graph #在创建node的时候指定对应的category，是一个个整数值 nodes = [ {"name": "木", "symbolSize": 10,"category":0}, {"name": "火", "symbolSize": 20,"category":1}, {"name": "土", "symbolSize": 30,"category":2}, {"name": "金", "symbolSize": 40,"category":3}, {"name": "水", "symbolSize": 50,"category":4}, ] #node的category通过index查找对应的类别 #不同类别的symbol会自动渲染成不同颜色 #还可以让不同类别节点设置不同的symbol符号 #如果类别symbol和节点symbol冲突，节点symbol优先 categories=[ {"name":"木类节点","symbol":"rect"}, {"name":"火类节点","symbol":"roundRect"}, {"name":"土类节点","symbol":"triangle"}, {"name":"金类节点","symbol":"diamond"}, {"name":"水类节点","symbol":"pin"}, ] #节点连线 links = [] for i in nodes: for j in nodes: links.append({"source": i.get("name"), "target": j.get("name")}) c = ( Graph() .add("a", nodes, links, categories=categories) ) c.render_notebook()
