networkx画图给后台传参

import networkx as nx
import matplotlib.pyplot as plt

fig = plt.figure()

# 下两行图上显示中文
mpl.rcParams['font.sans-serif'] = ['SimHei']

node_color = ['r']
data = {'张三':
            {'校友': ['李一','李二','李三','李四'],
             '同学': ['王一', '王二', '王三', '王四', '王五'],
              '同事': ['马一', '马二', '马三', '马四', '马五', '马六']
            }
        }
G = nx.Graph()
for k, v in data.items():
    G.add_node(k, id='0001')
    for m, n in v.items():
        if m == '校友':
            node_color.extend(['b'] * len(n))
            for i in n:
                G.add_node(i, id='0002')
                G.add_edge(k, i, weight=0.1)
        elif m == '同学':
            node_color.extend(['b'] * len(n))
            for i in n:
                G.add_node(i, id='0003')
                G.add_edge(k, i, weight=0.5)
        else:
            node_color.extend(['b'] * len(n))
            for i in n:
                G.add_node(i,id = '0003')
                G.add_edge(k,i,weight=1.0)



def on_press(event,relations):
    # print(event)
    # print('button=%d, x=%d, y=%d, xdata=%f, ydata=%f' %
    #       (event.button, event.x, event.y, event.xdata, event.ydata))
    if event.button == 1:
        for k,v in relations.items():
            if all([v[0][0]