如何让多个小部件按钮在Jupyter Notebook中执行不同的操作？

from ipywidgets import Button, VBox
from IPython.display import display

def action1(button):
    # 执行操作1的代码

def action2(button):
    # 执行操作2的代码

# 创建按钮并绑定对应的操作函数
button1 = Button(description='按钮1')
button1.on_click(action1)

button2 = Button(description='按钮2')
button2.on_click(action2)

# 将按钮放置在垂直布局的容器中
container = VBox([button1, button2])

# 显示容器和按钮
display(container)