使用networkx生成绘制节点的原始数据，使用matplotlib.pyplot进行绘制，使用了canvas.mpl_connect鼠标事件绑定，使图形根据数据更新动起来，直接上代码结果！鼠标左键点击新增节点；右键点击删除该节点下的其他节点，保留主节点；中间点击显示该节点详细信息。
#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Time : 2022/10/2 16:51
# @Author : singweek
# @File: Plt_Tree
import networkx as nx
import matplotlib.pyplot as plt
"""创建图"""
# G=nx.Graph()#无多重边无向图。忽略两个节点之间的多个边，允许节点自身成环。
# G = nx.DiGraph()#无多重边有向图
# G=nx.MultiGraph()#有多重边无向图，允许在成对的节点之间存在多个无向边。
# G=nx.MultiDiGraph()#有多重边有向图
# G = nx.davis_southern_women_graph()#构造关系图
"""新增节点边界以及删除节点边界"""
# G.add_node("start", subset=0)#添加节点
# G.add_nodes_from([1,2,3,4,5],weight=1)
# G.add_edges_from([('start','2'),('start','3')])#更新多个边界新信息，新增连接
# G.add_edge(('3', '5'))#更新单个边界信息
# G.remove_edges_from([('start','2'),('start','3')])#删除多个边界信息，start-2
# G.remove_edge(('start','2'))#删除单个边界信息
# G.remove_node()
# G.remove_nodes_from()
"""绘制节点显示"""
# nx.draw(G, pos, with_labels=True, node_color=node_color_l, edge_color=edge_color, node_size=1000) # ,with_labels=True
# # 展示
# plt.show()
def Get_Address_info(address):
"""
获取显示的数据，自己根据需求修改
return address_info_dict={'address':"address info"}
"""
address_info_dict={'start':"address info",'2':'adress 2','3':'adress3'}
if address in address_info_dict.keys():
return address_info_dict[address]
else:
return ''
def Get_Address_Node(address):
"""
获取节点数据，可以动态加载，自己根据需求修改
return address_node_dict={"node_cur":"start","node_pre":[1,2],"node_next":[3,4],"node_pre_color":'g',"node_next_color":'b'}
"""
Data_dict={'start':{"node_cur":'start','node_pre':['1-0','2-0','3-0'],'node_next':['1-1','2','1-3'],"node_pre_color":'g',"node_next_color":'b'},
'2':{"node_cur":'2','node_pre':['1','2-1','2-2'],'node_next':['2-1','2-2','3'],"node_pre_color":'g',"node_next_color":'b'},
'3':{"node_cur":'3','node_pre':['3-1','3-3','3-2'],'node_next':['3-4','3-5','4'],"node_pre_color":'g',"node_next_color":'b'}}
# address_node_dict={"node_cur":"start","node_pre":[1,2],"node_next":[3,4],"node_pre_color":'g',"node_next_color":'b'}
if address in Data_dict.keys():
return Data_dict[address]
else:
return {}
# 鼠标点击响应函数
def on_press(event,loc,error_xy=0.05):
"""
鼠标点击事件判断
event.button 1、2、3分别为鼠标左键、中键、右键点击
event.x, event.y 显示屏幕的坐标，从左上角开始计算
event.xdata, event.ydata 绘图中的坐标，从左上角开始计算，0-1范围内
loc 传入所有节点的参数及坐标
error_xy 点击坐标误差
"""
global Node_Line
print("点击坐标：",'button=%d, x=%d, y=%d, xdata=%f, ydata=%f' %(event.button, event.x, event.y, event.xdata, event.ydata))
print("节点坐标",loc)
x,y=event.xdata, event.ydata
for i in loc.keys():
tx,ty=loc[i]
if tx-error_xy<=x<=tx+error_xy and ty-error_xy<=y<=ty+error_xy:
print("点击节点：",i)
if event.button==2:#鼠标中键显示节点信息
address_info=Get_Address_info(i)#获取节点信息，以便用于绘制{'1':'address'}
if address_info!="":
Plt_labels(i,address=address_info)
plt.show()
if event.button==1:#鼠标左键点击，扩张节点
node_cur_dict=Get_Address_Node(i)#获取节点信息，用于绘制
if node_cur_dict!={}:
node_cur = node_cur_dict['node_cur']
node_pre = node_cur_dict['node_pre']
node_next = node_cur_dict['node_next']
node_pre_color = node_cur_dict['node_pre_color']
node_next_color = node_cur_dict['node_next_color']
Node_Line.append(node_cur)
Add_Refresh_plt(node_cur,node_pre,node_next,node_pre_color=node_pre_color,node_next_color=node_next_color)
if event.button==3:#鼠标右键点击，删除节点
node_cur_dict=Get_Address_Node(i)#获取节点信息，用于绘制
if node_cur_dict != {}:
node_cur = node_cur_dict['node_cur']
node_pre = node_cur_dict['node_pre']
node_next = node_cur_dict['node_next']
node_pre_color = node_cur_dict['node_pre_color']
node_next_color = node_cur_dict['node_next_color']
Delete_Refresh_plt(node_cur, node_pre, node_next,node_pre_color=node_pre_color,node_next_color=node_next_color)
def Add_Refresh_plt(node_cur,node_pre=[],node_next=[],node_pre_color='green',node_next_color='blue',edge_color='black'):
"""
新增节点，刷新画布
:param node_cur开始点击的节点
:param node_pre点击的上一节点[1,2,3]传入当前地址的节点
:param node_next点击的下一节点[1,2,3]当前节点的输出节点
# :param node_color节点颜色{1:'r',2:'y'}
:param node_pre_color新增节点颜色'r'
:param node_next_color新增节点颜色'y'
:param edge_color边界颜色'black'默认黑色，所有边界颜色都一样
"""
global cid,node_color,Node_Line,pos
pos_o = nx.spring_layout(G) # 获取绘制前节点位置
# if node_cur=="start":#暂时定位start
node_color.update({node_cur: 'red'}) # 修改点击节点颜色，默认红色
"""新增上一节点"""
for i in node_pre:
if i not in pos_o.keys():#如果新增节点不在之前绘制的节点内
G.add_node(i)#新增节点
node_color.update({i:node_pre_color})#新增节点颜色
for i in node_pre:
G.add_edge(node_cur, i)
"""新增下一节点"""
for i in node_next:
if i not in pos_o.keys():#如果新增节点不在之前绘制的节点内
G.add_node(i)#新增节点
node_color.update({i: node_next_color}) # 新增节点颜色
for i in node_next:
G.add_edge(node_cur,i)
pos = nx.spring_layout(G) # 获取新增节点位置
fig.canvas.mpl_disconnect(cid)#解除鼠标事件绑定
plt.clf()#
cid=fig.canvas.mpl_connect('button_press_event', lambda event: on_press(event, pos))#鼠标事件绑定
node_color_l=[node_color[i] for i in node_color.keys()]
nx.draw(G, pos, with_labels=True, node_color=node_color_l, edge_color=edge_color, node_size=1000) #绘制
plt.show()#显示
def Delete_Refresh_plt(node_cur,node_pre=[],node_next=[],node_pre_color='green',node_next_color='blue',edge_color='black'):
"""
删除节点，刷新画布
:param node_cur开始点击的节点
:param node_pre点击的上一节点[1,2,3]传入当前地址的节点
:param node_next点击的下一节点[1,2,3]当前节点的输出节点
# :param node_color节点颜色{1:'r',2:'y'}
:param node_pre_color新增节点颜色'r'
:param node_next_color新增节点颜色'y'
:param edge_color边界颜色'black'默认黑色，所有边界颜色都一样
"""
global cid,node_color,Node_Line,pos
# pos_o = nx.spring_layout(G) # 获取绘制前节点位置
# if node_cur=="start":#暂时定位start
node_color.update({node_cur: 'red'}) # 修改点击节点颜色，默认红色
# print("删除节点",node_pre,node_next)
for i in node_next:#删除下一个节点
if i not in Node_Line:
try:
G.remove_node(i)
del node_color[i]
except:
print("删除出错节点：", i)
for i in node_pre:#删除上一个节点
if i not in Node_Line:
try:
G.remove_node(i)
del node_color[i]
except:
print("删除出错节点：", i)
pos = nx.spring_layout(G) #获取新增节点位置
fig.canvas.mpl_disconnect(cid)#解除鼠标事件绑定
plt.clf()#
cid=fig.canvas.mpl_connect('button_press_event', lambda event: on_press(event, pos))#鼠标事件绑定
node_color_l=[node_color[i] for i in node_color.keys()]
nx.draw(G, pos, with_labels=True, node_color=node_color_l, edge_color=edge_color, node_size=1000) #绘制
plt.show()#显示
def Plt_labels(node,address="address",error_xy=0.2,font_color='magenta'):
"""
绘制节点标签
# :param pos 坐标{'start': array([0.27709403, 0.38102859])}
:param node 节点名称
:param address 绘制参数
:param error_xy 绘制位置调整
:param font_color 'magenta'
"""
# pos = nx.spring_layout(G) # 获取新增节点位置
global pos
xy=pos[node]
pos_label={'start':[xy[0],xy[1]-error_xy]}
node_label={'start':address}
nx.draw_networkx_labels(G, pos_label, labels=node_label,font_color=font_color)
if __name__ == '__main__':
start_node="start"
Node_Line = [start_node]
G = nx.Graph() # 无多重边无向图。忽略两个节点之间的多个边，允许节点自身成环。
G.add_node(start_node)
pos = nx.spring_layout(G)#获取节点位置
node_color={'start':'red'}#起始节点颜色
node_color_l=[node_color[i] for i in node_color.keys()]
edge_color='black'
fig = plt.figure()
#绑定鼠标事件
cid=fig.canvas.mpl_connect('button_press_event',lambda event:on_press(event,pos))
# networkx画关系图
nx.draw(G, pos, with_labels=True,node_color=node_color_l, edge_color=edge_color, node_size=1000) # ,with_labels=True
# 展示
plt.show()
本文章为转载内容，我们尊重原作者对文章享有的著作权。如有内容错误或侵权问题，欢迎原作者联系我们进行内容更正或删除文章。
