Как перерисовать график в matplotlib?

Question

[Степан]

Есть программа,которая рисует траекторию полёта снаряда. При запуске программы всё работает хорошо,но мне надо этот график перерисовывать.Пробовал очищать график,удалять - ничего не помогало

Подскажите,как перерисовать график?Функция draw1 должна как раз его перерисовывать

plt.ion()
V0=10 #Нач скорость
a=45 #Угол
m=1 #Масса
k=1 #Коэфф сопротивления воздуха

class ExampleApp(QtWidgets.QMainWindow, design.Ui_MainWindow):
    def __init__(self):
        super().__init__()
        self.setupUi(self) 

        m = PlotCanvas(self, width=7, height=5.5)
        m.move(400,10)

        self.pushButton.clicked.connect(self.draw1)

    def draw1(self):
        global V0
        V0=100
        PlotCanvas()
        
        
class PlotCanvas(FigureCanvas):

    def __init__(self, parent=None, width=5, height=4, dpi=100):
        fig = Figure(figsize=(width, height), dpi=dpi)
        FigureCanvas.__init__(self, fig)
        self.setParent(parent)
        rect = fig.patch 
        rect.set_facecolor('#19232d')
        fig.clear()
        FigureCanvas.setSizePolicy(self,
                QSizePolicy.Expanding,
                QSizePolicy.Expanding)
        FigureCanvas.updateGeometry(self)
        self.plot()

    def Lenght(self,V0, a, m, k): #Расчёт траектории (совокупность координат (x,y) )
        ...
        return x_m, y
    def plot(self):
        global V0, a, m, k,line1
        x_m, y = self.Lenght(V0, a, m, k)  #Передача данных в функцию,которая вернёт данные для построения графика
        ax = self.figure.add_subplot(111) 

       #Отрисовка
        ax.plot(x_m, y, color="#ffab03")
       
        self.draw()
        print(1)

def main():
    app = QtWidgets.QApplication(sys.argv)  
    window = ExampleApp()  
    app.setStyleSheet(qdarkstyle.load_stylesheet(qt_api='pyqt5'))
    window.show()  
    app.exec_()  

if __name__ == '__main__':  
    main()

Answer 1

[dmshar]

То, что вы хотите сделать называется "анимированный график".
Анимированный график в Matplotlib создается с помощью FuncAnimation из модуля анимации matplotlib.animation. Вот ее и используйте, а не изобретайте свой велосипед.

Comments

[Степан]

так я не хочу анимацию.Я хочу при нажатии на кнопку просто новый график

[dmshar]

Тогда просто. удаляете старый. Создаете новый и рисуете. Запускать весь ваш код и разбираться с QT, вашими непонятными расчетами расчетами и пр - особого желание вряд-ли у кого возникнет. желание нет.
Хотите конкретной помощи - выделите минимально воспроизводимый фрагмент кода. Который у вас "работает но не так, как вы ожидаете. А народ посмотрит и попробует вам помочь.
Для начала, попробуйте использовать

plt.close()

[Степан]

dmshar, код в вопросе сократил,оставил главное. plt.close() не помог(

[dmshar]

Степан,
Ну как это не помог?

import matplotlib.pyplot as plt
fig,ax = plt.subplots()
x=[1,2,3,4]
y=[6,5,4,3]
ax.scatter(x,y)
plt.show()
plt.close()

fig,ax = plt.subplots()
ax.plot(x,y)
plt.show()

Выполняйте скрип построчно и понаблюдайте, как сперва создается окошко, потом там рисуется скаттер-плот, потом окошечко закрывается, потом открывается новое и в нем рисуется уже plot-график. Это то, что вы простили. Осталось это обернуть в ваши функции и если нужно - циклы.

[Степан]

dmshar, вот ваш пример при запуске программы работает.Но мне надо как то менять график при нажатии на кнопку,в этом вся проблема: как там связать классы,функции и тд,чтобы нормально работало

[Степан]

dmshar, вот сейчас вставил ваш код в обработчик нажатия кнопки.Но питон открывает график в новом окне,как запихнуть его в одно окно?

Answer 2

[Viktor T2]

# --------- рисование графика -------------
figure.clf() # затереть старые графики
ax = figure.add_subplot(111) # новая область рисования графика

Comments

[Степан]

не помогло(

[Viktor T2]

Степан,

затем всю вашу последовательность рисования повторить с другим параметром,
это основная идея

[Viktor T2]

Степан,

[Степан]

Viktor T2, делал именно так,всё равно рисует лишь 1 раз(

[Степан]

Viktor T2,
Если я вставлю эту команду перед выводом,то всё стирается,то есть эта строчка работает,но в начале почему то не работает

[Viktor T2]

Степан,
разбирайтесь с последовательностью ваших команд, рисования
вставить clf перед добавлением add_subplot, или еще там что,
там ничего сложного нет
старый график убираем, новый добавляем и перерисовываем

[Viktor T2]

def draw1(self):
        global V0
        V0=100
        PlotCanvas()

PlotCanvas.plot() - Нет ?

[Степан]

Viktor T2, пишет ошибку:

plot() missing 1 required positional argument: 'self'

Если в скобки запихнуть self, то другая ошибка

'ExampleApp' object has no attribute 'Lenght'

[Viktor T2]

Степан, Не знаю, не секу в ООП :)
Разбирайтесь с вашими обьектами:
что и как вызывается

[Степан]

Viktor T2, эх,ну вот и я не секу в ООП ((

[Viktor T2]

Степан,
myplot = PlotCanvas() # Создаем экземпляр обьекта (из класса PlotCanvas)
myplot.plot() # Вызываем метод (процедуру)
как-то так
PS Когда обьект создается автоматически всегда вызывается __init__()
Разбирайтесь что у вас делает инит и что делает плот

[Степан]

Viktor T2, теперь ошибок нет) Но опять не перерисовывается( Мне кажется это из-за того,что мы создали новую фигуру,а при следующем запуске ссылку на эту фигуру потеряли и никак к ней не обратиться

[Viktor T2]

Степан,
Гоу бухать читать учебники