matplotlib 绘图问题
Posted
技术标签:
【中文标题】matplotlib 绘图问题【英文标题】:Issue with matplotlib plotting 【发布时间】:2020-02-12 07:29:15 【问题描述】:我目前正面临 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'mycurveomega')
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_())
【问题讨论】:
【参考方案1】:在将 Matplotlib 集成到 PyQt 中时,您永远不应该使用 matplotlib.pyplot
模块,因为它会创建自己的事件并维护图形和轴的列表。它与 Qt 冲突。
有一个很好的例子来说明如何在 PyQt here 中嵌入 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'mycurveomega')
self._canvas.draw()
app = QtWidgets.QApplication(sys.argv)
materialbrowser = MaterialBrowser()
sys.exit(app.exec_())
顺便说一句,我强烈建议使用 Qt 布局来布局您的小部件。
【讨论】:
以上是关于matplotlib 绘图问题的主要内容,如果未能解决你的问题,请参考以下文章