69 lines
3.6 KiB
Python
69 lines
3.6 KiB
Python
import Ui_MainWindow
|
||
from PyQt5.QtCore import QFile, QFileInfo, pyqtSlot, pyqtSignal
|
||
from PyQt5 import QtGui, QtWidgets, QtCore
|
||
from PyQt5.QtWidgets import QPushButton, QTableWidgetItem
|
||
from PyQt5.QtGui import QIcon, QPixmap
|
||
from data_set_read import Data_Read
|
||
import pandas as pd
|
||
import openpyxl as pyx
|
||
|
||
class MainWindows:
|
||
def __init__(self) -> None:
|
||
self.MainWindow = QtWidgets.QMainWindow()
|
||
self.ui = Ui_MainWindow.Ui_MainWindow()
|
||
self.ui.setupUi(self.MainWindow)
|
||
self.window_ui_init()
|
||
self.file_object_init()
|
||
self.head_num = 10
|
||
|
||
def window_ui_init(self):
|
||
self.MainWindow.setWindowTitle("Data Analysis By Python")
|
||
self.MainWindow.setWindowIcon(QIcon(QPixmap(":/src/image/icon.jpg")))
|
||
self.ui.progressBar_linear_regression.setValue(0)
|
||
|
||
def file_object_init(self):
|
||
self.ui.pushButton_filepath.clicked.connect(self.pushButton_filepath_clicked)
|
||
self.ui.pushButton_filepath_clear.clicked.connect(self.pushButton_filepath_clear_clicked)
|
||
self.ui.pushButton_linear_regression.clicked.connect(lambda checked: self.ui.stackedWidget.setCurrentIndex(0))
|
||
self.ui.pushButton_Kmeans.clicked.connect(lambda checked: self.ui.stackedWidget.setCurrentIndex(1))
|
||
|
||
def pushButton_filepath_clicked(self):
|
||
file_path = QtWidgets.QFileDialog.getOpenFileName(self.MainWindow, "Open File", ".", "All Files(*);;csv Files(*.csv);;Old Excel Files(*.xls);;New Excel Files(*.xlsx)")
|
||
print(file_path[0])
|
||
info = QFileInfo(file_path[0])
|
||
suffix = info.suffix()
|
||
if suffix == "csv":
|
||
self.data_read = Data_Read(file_path[0], suffix)
|
||
self.ui.lineEdit_filepath.setText(file_path[0])
|
||
elif suffix == "xls" or suffix == "xlsx":
|
||
self.data_read = Data_Read(file_path[0], suffix)
|
||
self.ui.lineEdit_filepath.setText(file_path[0])
|
||
|
||
if self.ui.stackedWidget.currentIndex() == 0:
|
||
self.tableWidget_linear_regression_show()
|
||
#data_show = self.data_read.get_head_data(self.head_num)
|
||
# self.tableWidget_linear_regression_show(data_show)
|
||
# if file_path is not None:
|
||
# self.ui.lineEdit_filepath.setText(file_path)
|
||
|
||
def tableWidget_linear_regression_show(self):
|
||
# self.ui.tableWidget_linear_regression.clear()
|
||
# # self.ui.tableWidget_linear_regression.setRowCount(data_frame.shape[0])
|
||
# # self.ui.tableWidget_linear_regression.setColumnCount(data_frame.shape[1])
|
||
# # for row_index, row in data_frame.iterrows():
|
||
# # for col_index, data in enumerate(row):
|
||
# # item = QTableWidgetItem(str(data))
|
||
# # self.ui.tableWidget_linear_regression.setItem(row_index, col_index, item)
|
||
self.ui.tableWidget_linear_regression.clear()
|
||
workbook = pyx.load_workbook(self.ui.lineEdit_filepath.text())
|
||
sheet = workbook.worksheets[0]
|
||
self.ui.tableWidget_linear_regression.setRowCount(self.head_num)
|
||
self.ui.tableWidget_linear_regression.setColumnCount(sheet.max_column)
|
||
for row in sheet.iter_rows(min_row=1, max_row=self.head_num, max_col=sheet.max_column, values_only=True):
|
||
print(row)
|
||
for col, data in enumerate(row, start=0):
|
||
item = QTableWidgetItem(str(data) if data is not None else "")
|
||
self.ui.tableWidget_linear_regression.setItem(row[0] - 1, col, item) # 注意行索引从0开始，但Excel从1开始
|
||
def pushButton_filepath_clear_clicked(self):
|
||
self.ui.lineEdit_filepath.clear()
|
||
|