时间戳

首页 > 解决方案 > matplotlib 绘图问题

python - matplotlib 绘图问题

问题描述

我目前正面临 matplotlib 与 PyQt5 结合的问题，我根本不明白。基本上，我只想在同一个图中绘制不同的曲线，由 QPushButton 触发。我在下面的最小工作示例中重新创建了问题。

期待：

每次我点击“绘图”按钮时，当前图形都会更新一行

结果：

第一个图可以正常工作，但是之后该图不再使用正确的数据进行更新。如果图形关闭并再次按下按钮，则绘图再次开始工作。如果代码更改为

fig = plt.figure()

也就是说，如果每次创建一个新图形时，它都会按预期工作。这是一个错误还是我以某种方式完全错了？

以下软件包用于最小工作示例：

matplotlib==3.1.3
PyQt5==5.14.1
pyqt5-tools==5.13.0.1.5

示例代码：

import matplotlib.pyplot as plt
import numpy as np
import sys
from PyQt5 import QtWidgets

class MaterialBrowser(QtWidgets.QMainWindow):

    def __init__(self):
        super(MaterialBrowser, self).__init__()
        self.setEnabled(True)
        self.setGeometry(0, 0, 543, 700)
        self.setMinimumSize(543, 400)
        self.save_button = QtWidgets.QPushButton(self)
        self.save_button.setGeometry(100, 100, 110, 32)
        self.save_button.setObjectName("plot_button")
        self.save_button.setText("Plot")
        self.save_button.clicked.connect(self.plot_bhcurve) 
        self.show()

    def plot_bhcurve(self):
        t = np.arange(0.0, 2.0, 0.01)
        omega = np.random.randint(2, 50)
        s = 1 + np.sin(2 * np.pi * omega* t)
        fig = plt.figure(num='MYFIGURE')
        ax = fig.gca()
        ax.plot(t, s, '.-', label=f'mycurve{omega}')
        ax.grid(True, which="both")
        ax.set_xlabel('Field Strength (A/m)')
        ax.set_ylabel('Flux Density (T)')
        ax.set_xlim(left=0)
        ax.set_ylim(bottom=0)
        plt.legend()
        plt.show()

app = QtWidgets.QApplication(sys.argv)
materialbrowser = MaterialBrowser()
sys.exit(app.exec_())

标签： pythonpython-3.xmatplotlibpyqt5

解决方案

将 Matplotlib 集成到 PyQt 时，您永远不应该使用该matplotlib.pyplot模块，因为它会创建它自己的事件并维护一个图形和轴列表。它与 Qt 冲突。

这里有一个关于如何在 PyQt 中嵌入 Matplotlib 的好例子。我已经在此基础上调整了您的示例。

import numpy as np
import sys
from PyQt5 import QtWidgets

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


class MaterialBrowser(QtWidgets.QMainWindow):

    def __init__(self):
        super(MaterialBrowser, self).__init__()
        self.setEnabled(True)
        self.setGeometry(0, 0, 543, 700)
        self.setMinimumSize(543, 400)

        self.main_widget = QtWidgets.QWidget()
        self.setCentralWidget(self.main_widget)

        self.main_layout = QtWidgets.QVBoxLayout()
        self.main_widget.setLayout(self.main_layout)

        self.save_button = QtWidgets.QPushButton(self)
        self.save_button.setGeometry(100, 100, 110, 32)
        self.save_button.setObjectName("plot_button")
        self.save_button.setText("Plot")
        self.save_button.clicked.connect(self.plot_bhcurve) 
        self.main_layout.addWidget(self.save_button)

        self._fig = Figure(figsize=(5, 3))
        self._canvas = FigureCanvas(self._fig)
        self.main_layout.addWidget(self._canvas)

        self._axes = self._fig.subplots()
        self._axes.grid(True, which="both")
        self._axes.set_xlabel('Field Strength (A/m)')
        self._axes.set_ylabel('Flux Density (T)')
        self._axes.set_xlim(left=0)
        self._axes.set_ylim(bottom=0)
        self._axes.legend()

        self.addToolBar(NavigationToolbar(self._canvas, self))

        self.show()



    def plot_bhcurve(self):
        print("plotting")
        t = np.arange(0.0, 2.0, 0.01)
        omega = np.random.randint(2, 50)
        s = 1 + np.sin(2 * np.pi * omega* t)

        self._axes.plot(t, s, '.-', label=f'mycurve{omega}')
        self._canvas.draw()


app = QtWidgets.QApplication(sys.argv)
materialbrowser = MaterialBrowser()
sys.exit(app.exec_())

顺便说一句，我强烈建议使用 Qt 布局来布局您的小部件。

推荐阅读