简介
在本实验中，你将学习如何使用 Matplotlib 创建一个简单的菜单。该菜单将包含一个项目列表，用户可以选择其中的每一项。然后，所选项目将触发一个回调函数，该函数将执行一些操作。
虚拟机使用提示
虚拟机启动完成后，点击左上角切换到“笔记本”标签，以访问 Jupyter Notebook 进行练习。
有时，你可能需要等待几秒钟让 Jupyter Notebook 完成加载。由于 Jupyter Notebook 的限制，操作验证无法自动化。
如果你在学习过程中遇到问题，请随时向 Labby 提问。课程结束后提供反馈，我们将立即为你解决问题。
导入库
首先，我们需要导入必要的库。在这种情况下，我们需要从matplotlib.transforms中导入matplotlib.pyplot、matplotlib.artist、matplotlib.patches和IdentityTransform。
import matplotlib.pyplot as plt
import matplotlib.artist as artist
import matplotlib.patches as patches
from matplotlib.transforms import IdentityTransform
定义 ItemProperties 类
接下来，我们定义一个ItemProperties类，该类将用于设置每个菜单项的属性。使用这个类，我们可以为每个菜单项设置字体大小、标签颜色、背景颜色和透明度值。
class ItemProperties:
def __init__(self, fontsize=14, labelcolor='black', bgcolor='yellow',
alpha=1.0):
self.fontsize = fontsize
self.labelcolor = labelcolor
self.bgcolor = bgcolor
self.alpha = alpha
定义 MenuItem 类
现在我们定义一个MenuItem类，该类将用于创建菜单中的每个项目。我们将图形、标签字符串、属性、悬停属性和选择时的回调函数作为参数传递给这个类。MenuItem类继承自artist.Artist类。
class MenuItem(artist.Artist):
padx = 5
pady = 5
def __init__(self, fig, labelstr, props=None, hoverprops=None,
on_select=None):
super().__init__()
self.set_figure(fig)
self.labelstr = labelstr
self.props = props if props is not None else ItemProperties()
self.hoverprops = (
hoverprops if hoverprops is not None else ItemProperties())
if self.props.fontsize!= self.hoverprops.fontsize:
raise NotImplementedError(
'support for different font sizes not implemented')
self.on_select = on_select
## Setting the transform to IdentityTransform() lets us specify
## coordinates directly in pixels.
self.label = fig.text(0, 0, labelstr, transform=IdentityTransform(),
size=props.fontsize)
self.text_bbox = self.label.get_window_extent(
fig.canvas.get_renderer())
self.rect = patches.Rectangle((0, 0), 1, 1) ## Will be updated later.
self.set_hover_props(False)
fig.canvas.mpl_connect('button_release_event', self.check_select)
def check_select(self, event):
over, _ = self.rect.contains(event)
if not over:
return
if self.on_select is not None:
self.on_select(self)
def set_extent(self, x, y, w, h, depth):
self.rect.set(x=x, y=y, width=w, height=h)
self.label.set(position=(x + self.padx, y + depth + self.pady/2))
self.hover = False
def draw(self, renderer):
self.rect.draw(renderer)
self.label.draw(renderer)
def set_hover_props(self, b):
props = self.hoverprops if b else self.props
self.label.set(color=props.labelcolor)
self.rect.set(facecolor=props.bgcolor, alpha=props.alpha)
def set_hover(self, event):
"""
Update the hover status of event and return whether it was changed.
"""
b, _ = self.rect.contains(event)
changed = (b!= self.hover)
if changed:
self.set_hover_props(b)
self.hover = b
return changed
定义 Menu 类
我们还需要定义一个Menu类，该类将用于创建菜单。我们将图形和菜单项列表作为参数传递给这个类。
class Menu:
def __init__(self, fig, menuitems):
self.figure = fig
self.menuitems = menuitems
maxw = max(item.text_bbox.width for item in menuitems)
maxh = max(item.text_bbox.height for item in menuitems)
depth = max(-item.text_bbox.y0 for item in menuitems)
x0 = 100
y0 = 400
width = maxw + 2*MenuItem.padx
height = maxh + MenuItem.pady
for item in menuitems:
left = x0
bottom = y0 - maxh - MenuItem.pady
item.set_extent(left, bottom, width, height, depth)
fig.artists.append(item)
y0 -= maxh + MenuItem.pady
fig.canvas.mpl_connect('motion_notify_event', self.on_move)
def on_move(self, event):
if any(item.set_hover(event) for item in self.menuitems):
self.figure.canvas.draw()
创建菜单
现在我们可以创建菜单了。我们创建一个新的图形并调整左边距。然后我们创建一个菜单项列表并将它们添加到菜单中。我们还为每个菜单项定义了一个回调函数，该函数将打印出所选的菜单项。
fig = plt.figure()
fig.subplots_adjust(left=0.3)
props = ItemProperties(labelcolor='black', bgcolor='yellow',
fontsize=15, alpha=0.2)
hoverprops = ItemProperties(labelcolor='white', bgcolor='blue',
fontsize=15, alpha=0.2)
menuitems = []
for label in ('open', 'close','save','save as', 'quit'):
def on_select(item):
print('you selected %s' % item.labelstr)
item = MenuItem(fig, label, props=props, hoverprops=hoverprops,
on_select=on_select)
menuitems.append(item)
menu = Menu(fig, menuitems)
plt.show()
总结
恭喜你！你已经成功地使用 Matplotlib 创建了一个简单的菜单。你已经学会了如何创建一个MenuItem类、一个Menu类，以及如何为每个菜单项定义属性。你还学会了如何为每个菜单项创建一个回调函数，该函数将执行某些操作。