matplotlib工具栏源码探析二（添加、删除内置工具项）探讨了工具栏内置工具项的管理，除了内置工具项，很多场景中需要自定义工具项，官方给出了案例https://matplotlib.org/gallery/user_interfaces/toolmanager_sgskip.html，主要基于matplotlib.backend_managers.ToolManager类实现，即使用工具栏管理器模式。

官方案例解析

下面对官方案例关键点做注释说明。

import matplotlib.pyplot as plt# 设置工具栏使用工具栏管理器模式plt.rcParams['toolbar'] = 'toolmanager'# 导入工具项的基类ToolBase和ToolToggleBasefrom matplotlib.backend_tools import ToolBase, ToolToggleBase# 因为工具项必须以类的形式添加，所以创建自定义基本工具项类，基类为ToolBaseclass ListTools(ToolBase): # 该工具项的功能为列出工具栏管理器管理的所有工具项 """List all the tools controlled by the `ToolManager`.""" # 设置默认快捷键和工具项描述 default_keymap = 'm' description = 'List Tools' # 定义工具项被触发时的动作 def trigger(self, *args, **kwargs): print('_' * 80) print("{0:12} {1:45} {2}".format( 'Name (id)', 'Tool description', 'Keymap')) print('-' * 80) # 获取工具栏管理器管理的所有工具项 tools = self.toolmanager.tools # 输出各个工具项 for name in sorted(tools): if not tools[name].description: continue keys = ', '.join(sorted(self.toolmanager.get_tool_keymap(name))) print("{0:12} {1:45} {2}".format( name, tools[name].description, keys)) print('_' * 80) print("Active Toggle tools") print("{0:12} {1:45}".format("Group", "Active")) print('-' * 80) for group, active in self.toolmanager.active_toggle.items(): print("{0:12} {1:45}".format(str(group), str(active)))# 基于ToolToggleBase创建自定义切换式工具项，切换式工具项在触发时会在生效和失效两种状态之间切换class GroupHideTool(ToolToggleBase): # 该工具项的功能为根据分组切换显示/隐藏数据元素 """Show lines with a given gid.""" # 设置默认快捷键和工具项描述 default_keymap = 'G' description = 'Show by gid' default_toggled = True # 构造函数的参数gid为数据元素的分组 def __init__(self, *args, gid, **kwargs): self.gid = gid super().__init__(*args, **kwargs) # 定义工具项生效时的方法 def enable(self, *args): self.set_lines_visibility(True) # 定义工具项失效时的方法 def disable(self, *args): self.set_lines_visibility(False) def set_lines_visibility(self, state): for ax in self.figure.get_axes(): for line in ax.get_lines(): if line.get_gid() == self.gid: line.set_visible(state) # 注意！在图像生成之后，修改图像中的元素必须重绘 self.figure.canvas.draw()fig = plt.figure()# 注意通过gid属性可以为数据元素分组plt.plot([1, 2, 3], gid='mygroup')plt.plot([2, 3, 4], gid='unknown')plt.plot([3, 2, 1], gid='mygroup')# 将自定义的工具项添加添加到工具栏管理器，格式为 工具项名称 工具项类 其他参数fig.canvas.manager.toolmanager.add_tool('List', ListTools)fig.canvas.manager.toolmanager.add_tool('Show', GroupHideTool, gid='mygroup')# 可以反复添加已存在的工具项# Add an existing tool to new group `foo`.# It can be added as many times as we wantfig.canvas.manager.toolbar.add_tool('zoom', 'foo')# 删除工具项# Remove the forward buttonfig.canvas.manager.toolmanager.remove_tool('forward')# 新添加到工具栏管理器的工具项还不能直接使用，需要通过toolbar对象添加到当前工具栏# 如果不将自定义的工具项添加到工具栏管理器，直接使用toolbar对象添加则会报错# 将自定义的工具项Show添加到内置的navigation组的特定位置（即组内第2个位置）# To add a custom tool to the toolbar at specific location inside# the navigation groupfig.canvas.manager.toolbar.add_tool('Show', 'navigation', 1)#fig.canvas.manager.toolbar.add_tool('List', 'navigation', 2)plt.show()

官方案例运行结果

运行后自定义的Show按钮处于生效状态，3条线全部显示。

点击Show按钮，使之处理失效状态，mygroup组的两条线不再显示。

由于案例中仅将List工具项添加到工具栏管理器，但是没有添加到工具栏中，因此List工具项未在工具栏中显示。但是List工具项的快捷键m是生效的，在界面上按快捷键m，控制台输出以下信息。

________________________________________________________________________________Name (id) Tool description Keymap--------------------------------------------------------------------------------List List Tools mShow Show by gid Gallnav Enable all axes toolmanager aback Back to previous view MouseButton.BACK, backspace, c, leftcopy Copy the canvas figure to clipboard cmd+c, ctrl+cfullscreen Toggle fullscreen mode ctrl+f, fgrid Toggle major grids ggrid_minor Toggle major and minor gridshelp Print tool list, shortcuts and description f1home Reset original view h, home, rnav Enable one axes toolmanager 1, 2, 3, 4, 5, 6, 7, 8, 9pan Pan axes with left mouse, zoom with right pquit Quit the figure cmd+w, ctrl+w, qquit_all Quit all figuressave Save the figure ctrl+s, ssubplots Configure subplotsxscale Toggle scale X axis L, kyscale Toggle scale Y axis lzoom Zoom to rectangle o________________________________________________________________________________Active Toggle toolsGroup Active--------------------------------------------------------------------------------default NoneNone {'Show'}

总结

matplotlib支持两种工具项：基本工具项（基类ToolBase）和切换式工具项（基类ToolToggleBase）。
基本工具项需要注意定义trigger方法，即工具项被触发时的动作。
切换式工具项需要注意定义enable和disable方法，即生效和失效两种状态的动作，如方法定义中牵扯到修改图像，需要注意重绘图像。
注意添加自定义工具项的流程！先将自定义的工具项添加到工具栏管理器，然后再添加到当前工具栏!内置工具项之所以不用添加到工具栏管理器是因为它们本身就已经添加在工具栏管理器！

到此这篇关于matplotlib工具栏源码探析三之添加、删除自定义工具项的文章就介绍到这了,更多相关matplotlib工具栏内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持！