Как я могу удалить строки matplot в tkinter?

Question

Я пытаюсь сделать GUI программу, которая отображает pandas dataframe (tkinter, matplotlib)

, но есть некоторые проблемы с назначением команд кнопок

, что я хочу сделать одна строка, видимая / невидимая, когда я нажимаю кнопку,

сначала я попытался назначить кнопки, используя для l oop, так что мне не нужно заботиться о количестве строк

, но когда я нажимаю кнопку, только последняя строка переключает состояние

, поэтому я ломаю l oop и присваиваю без l oop, и проверяю, как он работает.

как заставить первый метод работать ?

import numpy as np
import matplotlib.pyplot as plt
import tkinter as tk
from tkinter import ttk

from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg

# Sample Data
x = np.arange(100)
y1 = np.sin(x)
y2 = np.cos(x)

data = pd.DataFrame()
data['time'] = x
data['sin'] = y1
data['cos'] = y2


class graph():
    def __init__(self):
        self.win = tk.Tk()
        # Graph region
        self.frame1 = ttk.LabelFrame(self.win,text='Figure')
        self.frame1.grid(row=0,column=0,columnspan=2)

        self.fig,self.ax = plt.subplots()

        self.canvas = FigureCanvasTkAgg(self.fig,master=self.frame1)
        self.canvas.get_tk_widget().grid(row=0,column=0)
        self._read_data(data)

        # Make buttons using for loop
        # it doesn't work!!
        self.frame2= ttk.LabelFrame(self.win,text='graph handle')
        self.frame2.grid(row=1,column=0)

        for i,col in enumerate(self.pic_dic.keys()):
            ttk.Label(self.frame2,text=col).grid(row=i,column=0)
            ttk.Button(self.frame2,text='on/off Graph',command=lambda:self._onoff_pic(col)).grid(row=i,column=1)

        # Make buttons using without loop
        # but it works!!
        self.frame3= ttk.LabelFrame(self.win,text='graph handle (without for loop)')
        self.frame3.grid(row=1,column=1)
        ttk.Label(self.frame3,text='time').grid(row=0,column=0)
        ttk.Button(self.frame3,text='on/off',command=lambda:self._onoff_pic('time')).grid(row=0,column=1)
        ttk.Label(self.frame3,text='sin').grid(row=1,column=0)
        ttk.Button(self.frame3,text='on/off',command=lambda:self._onoff_pic('sin')).grid(row=1,column=1)
        ttk.Label(self.frame3,text='cos').grid(row=2,column=0)
        ttk.Button(self.frame3,text='on/off',command=lambda:self._onoff_pic('cos')).grid(row=2,column=1)

        self.canvas.draw()

    def _read_data(self,data):
        self.pic_dic = {}
        for col in data.columns:
            self.pic_dic[col] = {}
            self.pic_dic[col]['line'], = self.ax.plot(data['time'],data[col])
    def _onoff_pic(self,col):
        tmp = self.pic_dic[col]['line'].get_visible()   
        self.pic_dic[col]['line'].set_visible(not tmp)
        self.canvas.draw()        

a = graph()
a.win.mainloop()
a.win.quit()

спасибо!

Answer 1

В вашем коде:

for i,col in enumerate(self.pic_dic.keys()):
    ttk.Label(self.frame2,text=col).grid(row=i,column=0)
    ttk.Button(self.frame2,text='on/off Graph',command=lambda:self._onoff_pic(col)).grid(row=i,column=1)

значению col будет присвоено значение последней итерации после для l oop. Поэтому, когда лямбда вызывается при нажатии кнопки, будет использоваться окончательное значение col.

Вам необходимо использовать значение аргумента по умолчанию в лямбда-выражении:

for i, col in enumerate(self.pic_dic.keys()):
    ttk.Label(self.frame2, text=col).grid(row=i, column=0)
    ttk.Button(self.frame2, text='on/off Graph', command=lambda c=col: self._onoff_pic(c)).grid(row=i,column=1)

значение аргумента по умолчанию будет инициализировано при создании лямбды.