使用pyechart生成节点关系图

前言

前几天写过使用networkx生成节点关系图，但是，生成图的图中节点的分布，始终是不如人意，没找到合适的布局图参考这里。后来发现可以使用echart来实现类似的目的，就尝试了一下，不断调整之后得到了比较满意的效果。

代码如下

import matplotlib.pyplot as plt
import networkx as nx
import pandas as pd
import math
from collections import Counter

from pyecharts import options as opts
from pyecharts.charts import Graph
from pyecharts.globals import ThemeType

data = pd.read_excel('./源数据.xlsx', keep_default_na=False)
counter_list = Counter()
ori_list = {}
import numpy as np
set_list = []
a, b = 0, 0
for x, y in zip(data['source'], data['target']):
    if x != '' and y != '':
        set_list += [x, y]
        if (x, y) not in ori_list:
            ori_list[(x, y)] = 1
        else:
            ori_list[(x, y)] += 1
print(a, b)
ori_list = sorted(ori_list.items(), key=lambda x: x[1], reverse=False)
data_list = []
for k in ori_list:
    data_list.append(list(k[0]) + [k[1]])

length = 1
already_add_nodes = []
alread_add_edges = []
nodes_data = []
links_data = []
for k in data_list:
   for i in k[:2]:
       if i not in already_add_nodes:
       # 添加节点
           nodes_data.append(opts.GraphNode(name=str(i), symbol_size=15, value=str(i)))
           already_add_nodes.append(i)
   source, target, value = str(k[0]), str(k[1]), str(k[2])
   # 添加边
   if k[:2] not in alread_add_edges:
       if [k[1],k[0]] in alread_add_edges:
           # 神奇之处，curve=0.3，控制连线的扭曲度，一条线发生扭曲后，两条线就不重合了
           linestyle_opts=opts.LineStyleOpts(color="black",curve=0.5)
           label_opts = opts.LabelOpts(font_size=15, position="", color="black")
       else:
           linestyle_opts = opts.LineStyleOpts(color="black", curve=0.1)
           label_opts = opts.LabelOpts(font_size=15, position="", color="black")
       links_data.append(opts.GraphLink(source=source, target=target,
                                        value=value,
                                        linestyle_opts=linestyle_opts,
                                        label_opts=label_opts
                                        ))
       alread_add_edges.append(k[:2])
init_opts = opts.InitOpts(width="2000px",  # 图宽
                          height="1500px",  # 图高
                          renderer="svg",  # 渲染模式 svg 或 canvas，即 RenderType.CANVAS 或 RenderType.SVG
                          page_title="节点关系图",  # 网页标题
                          )
c = (
    Graph(init_opts)
        .add(
        "",
        nodes_data,
        links_data,
        repulsion=500,
        edge_symbol=[None, 'arrow'],
        edge_symbol_size=6,
        edge_label=opts.LabelOpts(
            is_show=True, position='middle', formatter="{c}"
        ),
    )
        .set_global_opts(
        title_opts=opts.TitleOpts(title="节点关系图")
    )
        .render("2020-连接数大于等于"+str(length)+"节点关系图.html")
)

效果图