如何向matplotlib工具栏上的用户定义按钮添加功能？

import matplotlib.pyplot as plt
from matplotlib.widgets import Button

fig, ax = plt.subplots()

def button_callback(event):
    # 在这里添加按钮功能的代码
    pass

button_ax = plt.axes([0.8, 0.05, 0.1, 0.075])  # 按钮的位置和大小
button = Button(button_ax, '自定义按钮')  # 按钮的文本
button.on_clicked(button_callback)  # 绑定按钮的回调函数

plt.show()