Das deutsche Python-Forum

Code: Alles auswählen

line, = ax.plot(np.random.rand(10))

Code: Alles auswählen

def update(data):
    line.set_ydata(data)
    return line,

Code: Alles auswählen

#!/usr/bin/env python
# coding: utf8
from __future__ import absolute_import, division, print_function
import sys
import numpy as np
from matplotlib import pyplot as plt
from matplotlib.animation import FuncAnimation
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
from PyQt4.QtGui import (
    QAction, QApplication, QHBoxLayout, QIcon, QMainWindow, QToolBar,
    QSizePolicy, QWidget
)


class AnimantionCanvas(FigureCanvas):

    def __init__(self, parent):
        figure, axes = plt.subplots()
        FigureCanvas.__init__(self, figure)
        self.setParent(parent)
        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
        self.updateGeometry()

        self.line, = axes.plot(np.random.rand(10))
        axes.set_ylim(0, 1)
        self.animation = FuncAnimation(
            figure, self.update_line, self.data_generator, interval=500
        )

    def update_line(self, data):
        self.line.set_ydata(data)
        return self.line,

    def data_generator(self):
        while True:
            yield np.random.rand(10)

    def set_pause(self, state):
        if state:
            self.animation.event_source.stop()
        else:
            self.animation.event_source.start()


class MainWindow(QMainWindow):

    def __init__(self):
        QMainWindow.__init__(self)
        self.setWindowTitle('Animation Pause Test')

        pause_action = QAction(
            QIcon.fromTheme('media-playback-pause'), 'pause', self
        )
        pause_action.setCheckable(True)

        tool_bar = QToolBar(self)
        self.addToolBar(tool_bar)
        tool_bar.addAction(pause_action)

        widget = QWidget(self)
        layout = QHBoxLayout(widget)

        canvas = AnimantionCanvas(widget)
        layout.addWidget(canvas)

        self.setCentralWidget(widget)

        pause_action.triggered.connect(canvas.set_pause)


def main():
    application = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(application.exec_())


if __name__ == '__main__':
    main()