Last active
March 24, 2023 07:56
-
-
Save pklaus/3e16982d952969eb8a9a to your computer and use it in GitHub Desktop.
Embedding Matplotlib Plots in PyQt5
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
# embedding_in_qt5.py --- Simple Qt4 application embedding matplotlib canvases | |
# | |
# Copyright (C) 2005 Florent Rougon | |
# 2006 Darren Dale | |
# with Updates from @boxcontrol | |
# <http://www.boxcontrol.net/embedding-matplotlib-plot-on-pyqt5-gui.html> | |
# <https://github.com/boxcontrol/matplotlibPyQt5> | |
# | |
# This file is an example program for matplotlib. It may be used and | |
# modified with no restriction; raw copies as well as modified versions | |
# may be distributed without limitation. | |
import sys | |
import random | |
import matplotlib | |
matplotlib.use("Qt5Agg") | |
from PyQt5 import QtCore | |
from PyQt5.QtWidgets import QApplication, QMainWindow, QMenu, QVBoxLayout, QSizePolicy, QMessageBox, QWidget | |
from numpy import arange, sin, pi | |
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas | |
from matplotlib.figure import Figure | |
class MyMplCanvas(FigureCanvas): | |
"""Ultimately, this is a QWidget (as well as a FigureCanvasAgg, etc.).""" | |
def __init__(self, parent=None, width=5, height=4, dpi=100): | |
fig = Figure(figsize=(width, height), dpi=dpi) | |
self.axes = fig.add_subplot(111) | |
# We want the axes cleared every time plot() is called | |
self.axes.hold(False) | |
self.compute_initial_figure() | |
# | |
FigureCanvas.__init__(self, fig) | |
self.setParent(parent) | |
FigureCanvas.setSizePolicy(self, | |
QSizePolicy.Expanding, | |
QSizePolicy.Expanding) | |
FigureCanvas.updateGeometry(self) | |
def compute_initial_figure(self): | |
pass | |
class MyStaticMplCanvas(MyMplCanvas): | |
"""Simple canvas with a sine plot.""" | |
def compute_initial_figure(self): | |
t = arange(0.0, 3.0, 0.01) | |
s = sin(2*pi*t) | |
self.axes.plot(t, s) | |
class MyDynamicMplCanvas(MyMplCanvas): | |
"""A canvas that updates itself every second with a new plot.""" | |
def __init__(self, *args, **kwargs): | |
MyMplCanvas.__init__(self, *args, **kwargs) | |
timer = QtCore.QTimer(self) | |
timer.timeout.connect(self.update_figure) | |
timer.start(1000) | |
def compute_initial_figure(self): | |
self.axes.plot([0, 1, 2, 3], [1, 2, 0, 4], 'r') | |
def update_figure(self): | |
# Build a list of 4 random integers between 0 and 10 (both inclusive) | |
l = [random.randint(0, 10) for i in range(4)] | |
self.axes.plot([0, 1, 2, 3], l, 'r') | |
self.draw() | |
class ApplicationWindow(QMainWindow): | |
def __init__(self): | |
QMainWindow.__init__(self) | |
self.setAttribute(QtCore.Qt.WA_DeleteOnClose) | |
self.setWindowTitle("application main window") | |
self.file_menu = QMenu('&File', self) | |
self.file_menu.addAction('&Quit', self.fileQuit, | |
QtCore.Qt.CTRL + QtCore.Qt.Key_Q) | |
self.menuBar().addMenu(self.file_menu) | |
self.help_menu = QMenu('&Help', self) | |
self.menuBar().addSeparator() | |
self.menuBar().addMenu(self.help_menu) | |
self.help_menu.addAction('&About', self.about) | |
self.main_widget = QWidget(self) | |
l = QVBoxLayout(self.main_widget) | |
sc = MyStaticMplCanvas(self.main_widget, width=5, height=4, dpi=100) | |
dc = MyDynamicMplCanvas(self.main_widget, width=5, height=4, dpi=100) | |
l.addWidget(sc) | |
l.addWidget(dc) | |
self.main_widget.setFocus() | |
self.setCentralWidget(self.main_widget) | |
self.statusBar().showMessage("All hail matplotlib!", 2000) | |
def fileQuit(self): | |
self.close() | |
def closeEvent(self, ce): | |
self.fileQuit() | |
def about(self): | |
QMessageBox.about(self, "About", | |
"""embedding_in_qt5.py example | |
Copyright 2015 BoxControL | |
This program is a simple example of a Qt5 application embedding matplotlib | |
canvases. It is base on example from matplolib documentation, and initially was | |
developed from Florent Rougon and Darren Dale. | |
http://matplotlib.org/examples/user_interfaces/embedding_in_qt4.html | |
It may be used and modified with no restriction; raw copies as well as | |
modified versions may be distributed without limitation.""" | |
) | |
if __name__ == '__main__': | |
app = QApplication(sys.argv) | |
aw = ApplicationWindow() | |
aw.setWindowTitle("PyQt5 Matplot Example") | |
aw.show() | |
#sys.exit(qApp.exec_()) | |
app.exec_() |
it shows error:
self.axes.hold(False)
AttributeError: 'AxesSubplot' object has no attribute 'hold'
it shows error:
self.axes.hold(False)
AttributeError: 'AxesSubplot' object has no attribute 'hold'
Error could be because you are using newer version of PyQT. Please change the line from self.axes.hold(False) to self.axes.clear() and it should work fine.
How can i add NavigationToolbar in this ?
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Gracias por compartir el presente código.