在PyQt5 GUI中嵌入matplotlib时出现内存泄漏

Question

我使用Qt Designer构建了一个Matplotlib GUI，基本上遵循this tutorial。我正在使用Python 3.5.2和pyqt 5.6.0。你可以看到下面的代码正在工作。然而，当更改图表时，系统使用的内存增加了，至少根据Windows 10任务管理器的说法是这样的。为了更好地重现这个问题，可以增加绘图命令中使用的随机值的数量。似乎rmmppl函数中的self.canvas.close()命令不足以实际释放已使用的内存。

如何防止内存使用量增加？

编辑：这里是GUI的屏幕截图。

from PyQt5 import QtCore, QtGui, QtWidgets
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt5agg import (
FigureCanvasQTAgg as FigureCanvas,
NavigationToolbar2QT as NavigationToolbar)
import window
import numpy as np


class Plotter(QtWidgets.QMainWindow, window.Ui_MainWindow):
    def __init__(self):
        super(Plotter, self).__init__()
        self.setupUi(self)
        self.fig_dict = {}    
        self.mplfigs.itemClicked.connect(self.changefig)

    def addmpl(self, fig):
        self.canvas = FigureCanvas(fig)
        self.mplvl.addWidget(self.canvas)
        self.canvas.draw()
        self.toolbar = NavigationToolbar(self.canvas,self.mplwindow, coordinates = True)
        self.mplvl.addWidget(self.toolbar)

    def rmmppl(self):
        self.mplvl.removeWidget(self.canvas)
        self.canvas.close()
        self.mplvl.removeWidget(self.toolbar)
        self.toolbar.close()

    def addfig(self, name, fig):
        self.fig_dict[name]=fig
        self.mplfigs.addItem(name)

    def changefig(self,item):
        text = item.text()
        self.rmmppl()
        self.addmpl(self.fig_dict[text])



def main():
    import sys

    fig1 = Figure()
    ax1f1= fig1.add_subplot(111)
    ax1f1.plot(np.random.rand(5))

    fig2 = Figure()
    ax1f2 = fig2.add_subplot(121)
    ax1f2.plot(np.random.rand(5))
    ax1f2 = fig2.add_subplot(122)
    ax1f2.plot(np.random.rand(10))    

    app=QtWidgets.QApplication(sys.argv)
    main=Plotter()
    main.addmpl(fig1)
    main.addfig('Figure 1', fig1)
    main.addfig('Figure 2', fig2)
    main.show()

    sys.exit(app.exec_())


if __name__ == '__main__':
    main()

这是基本的GUI结构window.py：

from PyQt5 import QtCore, QtGui, QtWidgets

class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(640, 432)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.horizontalLayout = QtWidgets.QHBoxLayout(self.centralwidget)
        self.horizontalLayout.setObjectName("horizontalLayout")
        self.mplwindow = QtWidgets.QWidget(self.centralwidget)
        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Preferred)
        sizePolicy.setHorizontalStretch(0)
        sizePolicy.setVerticalStretch(0)
        sizePolicy.setHeightForWidth(self.mplwindow.sizePolicy().hasHeightForWidth())
        self.mplwindow.setSizePolicy(sizePolicy)
        self.mplwindow.setObjectName("mplwindow")
        self.mplvl = QtWidgets.QVBoxLayout(self.mplwindow)
        self.mplvl.setContentsMargins(0, 0, 0, 0)
        self.mplvl.setObjectName("mplvl")
        self.horizontalLayout.addWidget(self.mplwindow)
        self.mplfigs = QtWidgets.QListWidget(self.centralwidget)
        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Maximum, QtWidgets.QSizePolicy.Expanding)
        sizePolicy.setHorizontalStretch(0)
        sizePolicy.setVerticalStretch(0)
        sizePolicy.setHeightForWidth(self.mplfigs.sizePolicy().hasHeightForWidth())
        self.mplfigs.setSizePolicy(sizePolicy)
        self.mplfigs.setMaximumSize(QtCore.QSize(200, 16777215))
        self.mplfigs.setMinimumSize(QtCore.QSize(200, 0))
        self.mplfigs.setObjectName("mplfigs")
        self.horizontalLayout.addWidget(self.mplfigs)
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 640, 31))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))

- Joscha Fregin

---

我有类似的问题，你找到解决方案了吗？ - bactone

是的，下面被接受的答案对我有用。 - Joscha Fregin

Answer 1

我已在Linux上执行了您的代码，并观察到相同的内存泄漏问题。 请尝试以下操作：

使用以下命令导入垃圾收集器：

import gc

然后修改您的rmmppl方法：

def rmmppl(self):
    self.canvas.close()
    self.canvas.deleteLater()
    self.toolbar.close()
    self.toolbar.deleteLater()
    gc.collect()

这是关于deleteLater的文档。请参考QObject类文档。

- J.Serra

---

我尝试过这个，但似乎不起作用。 - bactone