Python 如何使用blit有效地重新绘制多个matplotlib图

Python 如何使用blit有效地重新绘制多个matplotlib图,python,matplotlib,pyqt5,Python,Matplotlib,Pyqt5,我正在使用matplotlib和pyqt5将数据绘制成3个轴,用户可以在一个图中进行选择,这也将显示在其他两个图中。由于我使用的是大数据(多达1000万个点),因此绘图选择可能会很慢,尤其是当我需要绘制散点图时 我正在尝试使用matplotlib blit函数,但结果有一些问题。下面是一个最简单的例子 import matplotlib matplotlib.use('Qt5Agg') import numpy as np import sys from matplotlib.backends

我正在使用matplotlib和pyqt5将数据绘制成3个轴,用户可以在一个图中进行选择,这也将显示在其他两个图中。由于我使用的是大数据(多达1000万个点),因此绘图选择可能会很慢,尤其是当我需要绘制散点图时

我正在尝试使用matplotlib blit函数,但结果有一些问题。下面是一个最简单的例子

import matplotlib
matplotlib.use('Qt5Agg')
import numpy as np
import sys

from matplotlib.backends.qt_compat import QtCore, QtWidgets
from matplotlib.backends.backend_qt5agg import (FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
from matplotlib.figure import Figure


class ApplicationWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        self._main = QtWidgets.QWidget()
        self.setCentralWidget(self._main)
        layout = QtWidgets.QVBoxLayout(self._main)

        self.static_canvas = FigureCanvas(Figure(figsize=(10, 10)))
        layout.addWidget(self.static_canvas)
        layout.addWidget(NavigationToolbar(self.static_canvas, self))
        axes = self.static_canvas.figure.subplots(2, 1)
        self.ax1 = axes[0]
        self.ax2 = axes[1]
        self.ax1.cla()
        self.ax2.cla()

        button = QtWidgets.QPushButton('Click me!')
        button.clicked.connect(self.update_canvas_blit)
        layout.addWidget(button)
        # Fixing random state for reproducibility
        np.random.seed(19680801)

        # Create random data
        N = 50000
        x = np.random.rand(N)
        y = np.random.rand(N)

        self.ax1.scatter(x, y)
        self.points = self.ax1.scatter([],[], s=5, color='red')

        x = np.linspace(0, 1000, 100000)
        self.ax2.plot(x, np.sin(x))
        self.lines, = self.ax2.plot([],[], color='red')
        self.static_canvas.draw()

        self.background1 = self.static_canvas.copy_from_bbox(self.ax1.bbox)
        self.background2 = self.static_canvas.copy_from_bbox(self.ax2.bbox)

    def update_canvas_blit(self):
        N = 50
        x = np.random.rand(N)
        y = np.random.rand(N)

        self.static_canvas.restore_region(self.background1)
        self.points.set_offsets(np.c_[x,y])
        self.ax1.draw_artist(self.points)
        self.ax1.figure.canvas.blit(self.ax1.bbox)

        self.static_canvas.restore_region(self.background2)
        x = np.linspace(0, np.random.randint(500,1000), 1000)
        self.lines.set_data(x, np.sin(x))
        self.ax2.draw_artist(self.lines)
        self.ax2.figure.canvas.blit(self.ax2.bbox)

if __name__ == "__main__":
    qapp = QtWidgets.QApplication(sys.argv)
    app = ApplicationWindow()
    app.show()
    qapp.exec_()
单击按钮时,预期输出应仍然是相同的背景,随机点/线重新绘制。在某种程度上,这是正在发生的,但有一些奇怪的人工制品,看起来像是轴以某种方式相互吸引。但当我尝试将其保存到.png时,它将恢复到良好状态


问题是背景快照是在屏幕上尚未显示图形的时刻拍摄的。在这一点上,数字是10乘10英寸大。稍后,它将显示在qMain窗口中,并调整大小以适合小部件。
只有在这种情况发生后,才有必要拍摄背景快照

一种选择是使用1秒的计时器,然后才复制背景。这将如下所示

import numpy as np
import sys

from matplotlib.backends.qt_compat import QtCore, QtWidgets
from matplotlib.backends.backend_qt5agg import (FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
from matplotlib.figure import Figure


class ApplicationWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        self._main = QtWidgets.QWidget()
        self.setCentralWidget(self._main)
        layout = QtWidgets.QVBoxLayout(self._main)

        self.static_canvas = FigureCanvas(Figure(figsize=(10, 10)))
        layout.addWidget(self.static_canvas)
        layout.addWidget(NavigationToolbar(self.static_canvas, self))
        axes = self.static_canvas.figure.subplots(2, 1)
        self.ax1 = axes[0]
        self.ax2 = axes[1]
        self.ax1.cla()
        self.ax2.cla()

        button = QtWidgets.QPushButton('Click me!')
        button.clicked.connect(self.update_canvas_blit)
        layout.addWidget(button)
        # Fixing random state for reproducibility
        np.random.seed(19680801)

        # Create random data
        N = 50000
        x = np.random.rand(N)
        y = np.random.rand(N)

        self.ax1.scatter(x, y)
        self.points = self.ax1.scatter([],[], s=5, color='red')

        x = np.linspace(0, 1000, 100000)
        self.ax2.plot(x, np.sin(x))
        self.lines, = self.ax2.plot([],[], color='red')
        self.static_canvas.draw()

        self._later()


    def _later(self, evt=None):
        self.timer = self.static_canvas.new_timer(interval=1000)
        self.timer.single_shot = True
        self.timer.add_callback(self.update_background)
        self.timer.start()


    def update_background(self, evt=None):
        self.background1 = self.static_canvas.copy_from_bbox(self.ax1.bbox)
        self.background2 = self.static_canvas.copy_from_bbox(self.ax2.bbox)

    def update_canvas_blit(self):
        N = 50
        x = np.random.rand(N)
        y = np.random.rand(N)

        self.static_canvas.restore_region(self.background1)
        self.points.set_offsets(np.c_[x,y])
        self.ax1.draw_artist(self.points)
        self.ax1.figure.canvas.blit(self.ax1.bbox)

        self.static_canvas.restore_region(self.background2)
        x = np.linspace(0, np.random.randint(500,1000), 1000)
        self.lines.set_data(x, np.sin(x))
        self.ax2.draw_artist(self.lines)
        self.ax2.figure.canvas.blit(self.ax2.bbox)

if __name__ == "__main__":
    qapp = QtWidgets.QApplication(sys.argv)
    app = ApplicationWindow()
    app.show()
    qapp.exec_()

谢谢,这很有效!由于我在创建图形时没有在应用程序中使用
figsize
参数(我在这里只是作为一个示例使用),所以我可以从示例中删除它,它也可以工作。