使用单选按钮根据字典中的项目更改图形的绘图

Question

我有一本包含许多项目的字典。我想创建单个图形，其中单选按钮的数量等于字典中的项目数量。单选按钮选项将是字典的键，每次选择不同的单选按钮时，都会相应地绘制关联的值。例如：

from matplotlib.widgets import Slider, Button, RadioButtons
data = {'A' : [1,2,4,6,7,8,10],'B' : [1,2,4,6,4,3,7], 'C' : [3,2,1,6,2,3,4]}
axcolor = 'lightgoldenrodyellow'
rax = plt.axes([0.025, 0.5, 0.15, 0.15], facecolor=axcolor)
radio = RadioButtons(rax, data.keys(), active=0)
def chooseGraph(graph):
[plt.plot(data[names]) for names in data.keys()]
radio.on_clicked(chooseGraph)
plt.show()

当我执行此代码时，我得到单选按钮，但没有图形。当我选择一个按钮时，所有 3 个数据集都绘制在单选按钮的顶部，更改按钮只会更改颜色。

提前感谢您的帮助！

Answer 1

您需要创建一个用于绘制数据的轴，否则它将选择当前活动的轴，即带有单选按钮的轴。

import matplotlib.pyplot as plt
from matplotlib.widgets import RadioButtons
data = {'A' : [1,2,4,6,7,8,10],'B' : [1,2,4,6,4,3,7], 'C' : [3,2,1,6,2,3,4]}
fig, ax = plt.subplots()
ax.plot(data["A"])
rax = fig.add_axes([0.025, 0.5, 0.15, 0.15])
radio = RadioButtons(rax, sorted(data.keys()), active=0)
def chooseGraph(graph):
ax.clear()
ax.plot(data[graph])
fig.canvas.draw_idle()
radio.on_clicked(chooseGraph)
plt.show()