matplotlib

Matplotlib怎么实现根据下拉菜单选择显示不同数据

小亿
137
2024-05-21 14:19:35
栏目: 编程语言

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

0
看了该问题的人还看了