faculty_project/GraphWidget/GraphWidget.py

24 lines
714 B
Python

import matplotlib
matplotlib.use('Qt5Agg')
import Graph
from PyQt5 import QtWidgets
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg, NavigationToolbar2QT as NavigationToolbar
class GraphWidget(QtWidgets.QWidget):
def __init__(self, x, y, labels, mult_plots=False):
super().__init__()
graph = Graph.Graph(x,y,labels, mult_subplots=mult_plots)
sc = FigureCanvasQTAgg(graph.figure)
# Create toolbar, passing canvas as first parament, parent (self, the MainWindow) as second.
toolbar = NavigationToolbar(sc, self)
layout = QtWidgets.QVBoxLayout()
layout.addWidget(toolbar)
layout.addWidget(sc)
self.setLayout(layout)