要实现根据下拉菜单选择显示不同数据,可以使用Matplotlib中的interactive函数和widgets模块。下面是一个简单的示例代码:
import matplotlib.pyplot as plt
from matplotlib.widgets import Dropdown
# 数据
data1 = [1, 2, 3, 4, 5]
data2 = [5, 4, 3, 2, 1]
# 创建图形
fig, ax = plt.subplots()
plt.subplots_adjust(left=0.2, bottom=0.4)
# 绘制初始数据
line, = ax.plot(data1)
# 下拉菜单选项
options = ['Data 1', 'Data 2']
# 下拉菜单回调函数
def on_select(label):
if label == 'Data 1':
line.set_ydata(data1)
elif label == 'Data 2':
line.set_ydata(data2)
fig.canvas.draw()
# 创建下拉菜单
dropdown = Dropdown(plt.axes([0.1, 0.1, 0.3, 0.05]), options)
dropdown.on_clicked(on_select)
plt.show()
在这个示例中,我们创建了一个包含两组数据的下拉菜单,用户可以选择不同的数据来显示在图形中。当用户选择不同的选项时,回调函数on_select会根据选项的值更新图形中的数据,并重新绘制图形。