Отображение графика выбора опции со списком в PyQt GUI поверх существующего главного окна - PullRequest
0 голосов
/ 11 марта 2020

Таким образом, я пытаюсь настроить какой-то инструмент, который позволяет мне выбирать столбец в кадре данных, и когда я выбираю этот столбец в выпадающем списке, график, показывающий распределение этого столбца, должен отображаться в том же Окно. Я не знаю, как go об этом ...

Вот так выглядит мой комбинированный список:

enter image description here

I нужно уметь показывать график (распределение) в том же окне.

Как мне go об этом?

1 Ответ

1 голос
/ 11 марта 2020

Вот пример, иллюстрирующий использование поля со списком для отображения данных в столбце во фрейме данных в том же окне.

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()
...