要实现根据下拉菜单选择显示不同数据,可以使用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会根据选项的值更新图形中的数据,并重新绘制图形。