-- coding: utf-8 --

""" Module implementing MainWindow. """

from PyQt5.QtCore import pyqtSlot from PyQt5.QtWidgets import QMainWindow, QApplication

from PyQt5 import QtCore, QtGui, QtWidgets

class Ui_MainWindow(object): def setupUi(self, MainWindow): MainWindow.setObjectName("MainWindow") MainWindow.resize(800, 600) self.centralWidget = QtWidgets.QWidget(MainWindow) self.centralWidget.setObjectName("centralWidget") self.pushButton = QtWidgets.QPushButton(self.centralWidget) self.pushButton.setGeometry(QtCore.QRect(680, 60, 75, 23)) self.pushButton.setObjectName("pushButton") self.pushButton_2 = QtWidgets.QPushButton(self.centralWidget) self.pushButton_2.setGeometry(QtCore.QRect(680, 170, 75, 23)) self.pushButton_2.setObjectName("pushButton_2") self.pandastablewidget = DataTableWidget(self.centralWidget) self.pandastablewidget.setGeometry(QtCore.QRect(10, 30, 591, 331)) self.pandastablewidget.setStyleSheet("") self.pandastablewidget.setObjectName("pandastablewidget") MainWindow.setCentralWidget(self.centralWidget)

    self.retranslateUi(MainWindow)
    QtCore.QMetaObject.connectSlotsByName(MainWindow)

def retranslateUi(self, MainWindow):
    _translate = QtCore.QCoreApplication.translate
    MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
    self.pushButton.setText(_translate("MainWindow", "数据初始化"))
    self.pushButton_2.setText(_translate("MainWindow", "保存数据"))

from qtpandas.views.DataTableView import DataTableWidget

from qtpandas.models.DataFrameModel import DataFrameModel import pandas as pd

class MainWindow(QMainWindow, Ui_MainWindow): """ Class documentation goes here. """

def __init__(self, parent=None):
    """
    Constructor

    @param parent reference to the parent widget
    @type QWidget
    """
    super(MainWindow, self).__init__(parent)
    self.setupUi(self)

    '''初始化pandasqt'''
    widget = self.pandastablewidget
    widget.resize(600, 500)  # 如果对部件尺寸大小不满意可以在这里设置

    self.model = DataFrameModel()  # 设置新的模型
    widget.setViewModel(self.model)

    self.df = pd.read_excel(r'./data/fund_data.xlsx', encoding='gbk')
    self.df_original = self.df.copy()  # 备份原始数据
    self.model.setDataFrame(self.df)

@pyqtSlot()
def on_pushButton_clicked(self):
    """
    初始化pandas
    """
    self.model.setDataFrame(self.df_original)

@pyqtSlot()
def on_pushButton_2_clicked(self):
    """
    保存数据
    """
    self.df.to_excel(r'./data/fund_data_new.xlsx')

if name == "main": import sys

app = QApplication(sys.argv)
ui = MainWindow()
ui.show()
sys.exit(app.exec_())