如何获取单选按钮以更新直方图Matplotlib Python

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

fig, ax = plt.subplots()
data = [1, 2, 3, 3, 3, 4, 4, 5, 5, 5]
ax.hist(data, bins=5)

def update_hist(label):
    # 根据选择的单选按钮标签更新直方图
    if label == '全部数据':
        ax.hist(data, bins=5)
    elif label == '部分数据':
        ax.hist(data[:5], bins=5)
    else:
        ax.hist(data[5:], bins=5)
    plt.draw()

# 创建单选按钮对象并指定标签和位置
radio = RadioButtons(ax, ['全部数据', '部分数据', '剩余数据'])
# 将更新函数与单选按钮的“被点击”事件关联起来
radio.on_clicked(update_hist)

plt.show()