[英]Display a graph on selecting a combobox option on PyQt GUI , over an existing mainwindow
下面是一個示例,用於說明如何使用組合框在同一窗口中的數據框中繪制列中的數據。
from PyQt5 import QtWidgets
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
import matplotlib.pyplot as plt
import pandas as pd
import os
class Widget(QtWidgets.QWidget):
def __init__(self, parent=None):
super().__init__(parent)
self.combo = QtWidgets.QComboBox()
self.combo.addItem("Choose a field")
self.button = QtWidgets.QPushButton('Read csv file')
# axes and widget for plotting and displaying the figure
self.fig, self.ax = plt.subplots()
self.figure_widget = FigureCanvas(self.fig)
plt.tight_layout()
# set up layout
vlayout = QtWidgets.QVBoxLayout(self)
hlayout = QtWidgets.QHBoxLayout()
hlayout.addWidget(self.combo)
hlayout.addWidget(self.button)
hlayout.addStretch(2)
vlayout.addLayout(hlayout)
vlayout.addWidget(self.figure_widget)
self.button.clicked.connect(self.read_data)
self.combo.currentTextChanged.connect(self.field_changed)
def read_data(self):
dialog = QtWidgets.QFileDialog(self, directory=os.curdir, caption='Open data file' )
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptOpen)
dialog.setFileMode(QtWidgets.QFileDialog.ExistingFile)
if dialog.exec():
# read data and add columns to combo box
file = dialog.selectedFiles()[0]
self.data = pd.read_csv(file)
self.combo.clear()
self.combo.addItem("Choose a field")
for field in self.data.columns:
self.combo.addItem(field)
self.combo.setCurrentIndex(0)
def field_changed(self, field):
self.ax.clear()
if field in self.data.columns:
self.data.plot(y=field, ax=self.ax)
self.ax.set_ylabel(field)
self.ax.set_xlabel('index')
plt.tight_layout()
self.fig.canvas.draw()
if __name__ == '__main__':
app = QtWidgets.QApplication([])
widget = Widget()
widget.show()
app.exec()
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.