Использование панели навигации вместе с функцией анимации - PullRequest
Новая
       80

Использование панели навигации вместе с функцией анимации

0 голосов
/ 13 апреля 2020

Я пытаюсь запустить функцию Timedanimation в matplotlib в подклассе. Следующий код объясняет мою проблему на простом примере. Это показывает фигуру с маленьким красным кружком, бегущим по круговой дорожке. Блит это правда. 

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.lines import Line2D
import matplotlib.animation as animation


class SubplotAnimation(animation.TimedAnimation):
    def __init__(self):
        fig = plt.figure()
        ax1 = fig.add_subplot(1, 1, 1)

        self.t = np.linspace(0, 80, 400)
        self.x = np.cos(2 * np.pi * self.t / 10.)
        self.y = np.sin(2 * np.pi * self.t / 10.)
        self.z = 10 * self.t

        ax1.set_xlabel('x')
        ax1.set_ylabel('y')
        self.line1 = Line2D([], [], color='black')
        self.line1a = Line2D([], [], color='red', linewidth=2)
        self.line1e = Line2D(
            [], [], color='red', marker='o', markeredgecolor='r')
        ax1.add_line(self.line1)
        ax1.add_line(self.line1a)
        ax1.add_line(self.line1e)
        ax1.set_xlim(-3, 3)
        ax1.set_ylim(-4, 4)
        ax1.set_aspect('equal', 'datalim')

        animation.TimedAnimation.__init__(self, fig, interval=50, blit=True)

    def _draw_frame(self, framedata):
        i = framedata
        head = i - 1
        head_slice = (self.t > self.t[i] - 1.0) & (self.t < self.t[i])

        self.line1.set_data(self.x[:i], self.y[:i])
        self.line1a.set_data(self.x[head_slice], self.y[head_slice])
        self.line1e.set_data(self.x[head], self.y[head])

        self._drawn_artists = [self.line1, self.line1a, self.line1e]

    def new_frame_seq(self):
        return iter(range(self.t.size))

    def _init_draw(self):
        lines = [self.line1, self.line1a, self.line1e]
        for l in lines:
            l.set_data([], [])

ani = SubplotAnimation()
# ani.save('test_sub.mp4')
plt.show()

Теперь, если я немного поиграюсь с панелью инструментов навигации, как, например, увеличение или уменьшение масштаба, график предыдущего кадра до увеличения остается на графике следующим образом after zooming on the small circle.

Рамка остается на графике, пока я не разверну / сверну окно. Эта проблема случайная, она появляется не каждый раз. Кто-нибудь может подсказать мне, что я делаю неправильно? Заранее спасибо.

...