如何在fig.canvas.mpl_connect(‘key_press_event’,on_key)中传递参数？

在使用matplotlib库绘制图形时，我们有时会需要在键盘按下某特定键时触发相应的事件。这时我们通常会使用matplotlib的fig.canvas.mpl_connect(‘key_press_event’,on_key)来实现。

比如，我们要在按下键盘上的“s”键时，将当前图像保存为png格式的文件，我们可以这样实现：

更多Python教程，请阅读：Python 教程

import matplotlib.pyplot as plt

fig, ax = plt.subplots()

def on_key(event):
    if event.key == 's':
        fig.savefig('output.png')

fig.canvas.mpl_connect('key_press_event', on_key)

plt.show()

上述代码中，我们定义了一个on_key函数用来处理键盘按键事件，并将其传递给了fig.canvas.mpl_connect()函数，表示将on_key函数与按下键盘事件进行绑定。

其中，event.key 表示当前事件所对应的键位。在此处，我们判断当键位为“s”时，执行保存图像操作。否则，什么也不做。

但是，有时候我们在事件处理函数中需要使用到一些额外的参数，该怎么办呢？

这时，我们可以借助于python的functools.partial函数，进行事件函数的包装，将额外的参数传递给事件函数。

比如，下面的代码中，我们希望在按下“s”键时，将图像保存为指定文件名的png格式文件。这时，我们可以使用functools.partial函数将文件名这个参数传递给事件处理函数：

import matplotlib.pyplot as plt
from functools import partial

fig, ax = plt.subplots()

def on_key(event, filename):
    if event.key == 's':
        fig.savefig(filename)

fig.canvas.mpl_connect('key_press_event', partial(on_key, filename='output.png'))

plt.show()

其中，partial函数接收一个函数和若干个参数作为输入，返回一个新的函数，该新函数会固定原函数的部分参数，并将剩余的参数作为输入，最终返回原函数的调用结果。在这里，我们使用partial函数将on_key函数的第二个参数（即保存文件名）固定为“output.png”，并将其作为第一个参数传递给fig.canvas.mpl_connect()函数，表示将包含 filename 参数的 on_key 函数与键盘事件进行绑定。

这样，当我们按下“s”键时，on_key函数就会自动调用，保存图像到指定文件名的png格式文件中。