devbox
小小林熬夜学编程
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

[PyQt5]打开Excel、CSV，tableWidget转换为dataframe，listWidget添加选项，for循环列表推导式--list/dict_pyqt excel

作者：小小林熬夜学编程 | 2024-04-14 12:10:21

赞

踩

pyqt excel

untitled.py

  1. # -*- coding: utf-8 -*-
  2.  
  3. # Form implementation generated from reading ui file 'untitled.ui'
  4. #
  5. # Created by: PyQt5 UI code generator 5.15.9
  6. #
  7. # WARNING: Any manual changes made to this file will be lost when pyuic5 is
  8. # run again.  Do not edit this file unless you know what you are doing.
  9.  
  10.  
  11. from PyQt5 import QtCore, QtGui, QtWidgets
  12.  
  13.  
  14. class Ui_Form(object):
  15.     def setupUi(self, Form):
  16.         Form.setObjectName("Form")
  17.         Form.resize(640, 480)
  18.         self.verticalLayout_3 = QtWidgets.QVBoxLayout(Form)
  19.         self.verticalLayout_3.setObjectName("verticalLayout_3")
  20.         self.horizontalLayout_2 = QtWidgets.QHBoxLayout()
  21.         self.horizontalLayout_2.setObjectName("horizontalLayout_2")
  22.         self.pushButton_excel = QtWidgets.QPushButton(Form)
  23.         self.pushButton_excel.setObjectName("pushButton_excel")
  24.         self.horizontalLayout_2.addWidget(self.pushButton_excel)
  25.         self.pushButton_csv = QtWidgets.QPushButton(Form)
  26.         self.pushButton_csv.setObjectName("pushButton_csv")
  27.         self.horizontalLayout_2.addWidget(self.pushButton_csv)
  28.         self.verticalLayout_3.addLayout(self.horizontalLayout_2)
  29.         self.horizontalLayout = QtWidgets.QHBoxLayout()
  30.         self.horizontalLayout.setObjectName("horizontalLayout")
  31.         self.verticalLayout_2 = QtWidgets.QVBoxLayout()
  32.         self.verticalLayout_2.setObjectName("verticalLayout_2")
  33.         self.tableWidget = QtWidgets.QTableWidget(Form)
  34.         self.tableWidget.setObjectName("tableWidget")
  35.         self.tableWidget.setColumnCount(0)
  36.         self.tableWidget.setRowCount(0)
  37.         self.verticalLayout_2.addWidget(self.tableWidget)
  38.         self.pushButton_tW2df = QtWidgets.QPushButton(Form)
  39.         self.pushButton_tW2df.setObjectName("pushButton_tW2df")
  40.         self.verticalLayout_2.addWidget(self.pushButton_tW2df)
  41.         self.horizontalLayout.addLayout(self.verticalLayout_2)
  42.         self.verticalLayout = QtWidgets.QVBoxLayout()
  43.         self.verticalLayout.setObjectName("verticalLayout")
  44.         self.listWidget = QtWidgets.QListWidget(Form)
  45.         self.listWidget.setObjectName("listWidget")
  46.         self.verticalLayout.addWidget(self.listWidget)
  47.         self.pushButton_addLW = QtWidgets.QPushButton(Form)
  48.         self.pushButton_addLW.setObjectName("pushButton_addLW")
  49.         self.verticalLayout.addWidget(self.pushButton_addLW)
  50.         self.pushButton_fL = QtWidgets.QPushButton(Form)
  51.         self.pushButton_fL.setObjectName("pushButton_fL")
  52.         self.verticalLayout.addWidget(self.pushButton_fL)
  53.         self.pushButton_fD = QtWidgets.QPushButton(Form)
  54.         self.pushButton_fD.setObjectName("pushButton_fD")
  55.         self.verticalLayout.addWidget(self.pushButton_fD)
  56.         self.horizontalLayout.addLayout(self.verticalLayout)
  57.         self.horizontalLayout.setStretch(0, 3)
  58.         self.horizontalLayout.setStretch(1, 1)
  59.         self.verticalLayout_3.addLayout(self.horizontalLayout)
  60.  
  61.         self.retranslateUi(Form)
  62.         QtCore.QMetaObject.connectSlotsByName(Form)
  63.  
  64.     def retranslateUi(self, Form):
  65.         _translate = QtCore.QCoreApplication.translate
  66.         Form.setWindowTitle(_translate("Form", "Form"))
  67.         self.pushButton_excel.setText(_translate("Form", "打开Excel"))
  68.         self.pushButton_csv.setText(_translate("Form", "打开CSV"))
  69.         self.pushButton_tW2df.setText(_translate("Form", "tableWidget转换为dataframe"))
  70.         self.pushButton_addLW.setText(_translate("Form", "listWidget添加选项"))
  71.         self.pushButton_fL.setText(_translate("Form", "for循环列表推导式--list"))
  72.         self.pushButton_fD.setText(_translate("Form", "for循环列表推导式--dict"))

主代码： 

  1. import sys
  2. import pandas as pd
  3.  
  4. from PyQt5.QtWidgets import *
  5.  
  6. from untitled import Ui_Form
  7.  
  8.  
  9. class MyMainWindow(QWidget, Ui_Form):  # 继承 QWidget类和 Ui_Form界面类
  10.  
  11.     def __init__(self, parent=None):
  12.         super(MyMainWindow, self).__init__(parent)  # 初始化父类
  13.         self.setupUi(self)  # 继承 Ui_Form 界面类
  14.  
  15.         # 设置选择模式为多选模式
  16.         # self.listWidget.setSelectionMode(QListWidget.ExtendedSelection)
  17.         self.listWidget.setSelectionMode(QAbstractItemView.MultiSelection)
  18.  
  19.         # ********** 打开Excel
  20.         self.pushButton_excel.clicked.connect(self.openExcel)
  21.         self.pushButton_excel.clicked.connect(self.creatExcelTable)
  22.  
  23.         # ********** 打开CSV
  24.         self.pushButton_csv.clicked.connect(self.openCSV)
  25.         self.pushButton_csv.clicked.connect(self.creatCsvTable)
  26.  
  27.         # ********** 将tableWidget转换为dataframe
  28.         self.pushButton_tW2df.clicked.connect(self.tW2df)
  29.  
  30.         # ********** 设置up_listWidget选项
  31.         self.pushButton_addLW.clicked.connect(self.addlistWidgetItem)
  32.  
  33.         # ********** for循环列表推导式--list
  34.         self.pushButton_fL.clicked.connect(self.forLoopListDerivationFormula_list)
  35.  
  36.         # ********** for循环列表推导式--dict
  37.         self.pushButton_fD.clicked.connect(self.forLoopListDerivationFormula_dict)
  38.  
  39.     # 打开Excel
  40.     def openExcel(self):
  41.         openXlsName = QFileDialog.getOpenFileName(self, '选择Excel文件', '',
  42.                                                   'Excel files(*.xlsx , *.xls)')
  43.  
  44.         global excelPath
  45.  
  46.         excelPath = openXlsName[0]
  47.         # print("Excel路径：", excelPath)
  48.  
  49.     def creatExcelTable(self):
  50.         # print("Excel路径：", excelPath)
  51.         try:
  52.             # 读取Excel，将内容转写进表格
  53.             inputTable = pd.read_excel(excelPath)
  54.             # print(inputTable)
  55.  
  56.             # 设置表格行数与列数
  57.             inputTableRows = inputTable.shape[0]
  58.             inputTableColunms = inputTable.shape[1]
  59.             # print(inputTableRows)
  60.             # print(inputTableColunms)
  61.             self.tableWidget.setRowCount(inputTableRows)
  62.             self.tableWidget.setColumnCount(inputTableColunms)
  63.  
  64.             # 表头设置
  65.             inputTableHeader = inputTable.columns.values.tolist()
  66.             # print(inputTableHeader)
  67.             self.tableWidget.setHorizontalHeaderLabels(inputTableHeader)
  68.  
  69.             for i in range(inputTableRows):
  70.                 for j in range(inputTableColunms):
  71.                     self.tableWidget.setItem(i, j, QTableWidgetItem(str(inputTable.iloc[i, j])))
  72.  
  73.             # 设置表格不可编辑
  74.             self.tableWidget.setEditTriggers(QAbstractItemView.NoEditTriggers)
  75.  
  76.         except Exception as e:
  77.             QMessageBox.critical(self, "错误", f"读取Excel文件时出错。\n{e}")
  78.  
  79.     # 打开CSV
  80.     def openCSV(self):
  81.         openCsvName = QFileDialog.getOpenFileName(self, '选择CSV文件', '',
  82.                                                   'CSV files(*.csv)')
  83.  
  84.         global csvPath
  85.  
  86.         csvPath = openCsvName[0]
  87.         # print("Csv路径：", csvPath)
  88.  
  89.     def creatCsvTable(self):
  90.         # print("Csv路径：", csvPath)
  91.         try:
  92.             # 读取CSV，将内容转写进表格
  93.             inputTable = pd.read_csv(csvPath)
  94.             # print(inputTable)
  95.  
  96.             # 设置表格行数与列数
  97.             inputTableRows = inputTable.shape[0]
  98.             inputTableColunms = inputTable.shape[1]
  99.             # print(inputTableRows)
  100.             # print(inputTableColunms)
  101.             self.tableWidget.setRowCount(inputTableRows)
  102.             self.tableWidget.setColumnCount(inputTableColunms)
  103.  
  104.             # 表头设置
  105.             inputTableHeader = inputTable.columns.values.tolist()
  106.             # print(inputTableHeader)
  107.             self.tableWidget.setHorizontalHeaderLabels(inputTableHeader)
  108.  
  109.             for i in range(inputTableRows):
  110.                 for j in range(inputTableColunms):
  111.                     self.tableWidget.setItem(i, j, QTableWidgetItem(str(inputTable.iloc[i, j])))
  112.  
  113.             # 设置表格不可编辑
  114.             self.tableWidget.setEditTriggers(QAbstractItemView.NoEditTriggers)
  115.  
  116.         except Exception as e:
  117.             QMessageBox.critical(self, "错误", f"读取CSV文件时出错。\n{e}")
  118.  
  119.     def addlistWidgetItem(self):
  120.         # 清空选项框(必需)
  121.         self.listWidget.clear()
  122.  
  123.         # 返回主窗口表格列数
  124.         numberCol = self.tableWidget.columnCount()
  125.         # print(numberCol)
  126.         listWidgeDt = []
  127.         # 获取表格表头内容
  128.         for c in range(numberCol):
  129.             headerItem = self.tableWidget.horizontalHeaderItem(c)
  130.             # print(headerItem)
  131.             if headerItem != None:
  132.                 headerText = headerItem.text()
  133.                 # print(headerText)
  134.                 listWidgeDt.append(headerText)
  135.         # print(listWidgeDt)
  136.  
  137.         self.listWidget.addItems(listWidgeDt)
  138.         self.listWidget.selectAll()  # 设置全选
  139.  
  140.     def tableWidget2dataframe(self, tableWidget):
  141.         data = []
  142.         for r in range(self.tableWidget.rowCount()):
  143.             row = []
  144.             for c in range(self.tableWidget.columnCount()):
  145.                 item = self.tableWidget.item(r, c)
  146.                 if item is not None:
  147.                     row.append(item.text())
  148.                 else:
  149.                     row.append('')
  150.             data.append(row)
  151.         df = pd.DataFrame(data)
  152.         return df
  153.  
  154.     def tW2df(self):
  155.         print(self.tableWidget2dataframe(self.tableWidget))
  156.  
  157.     def forLoopListDerivationFormula_list(self):
  158.         # 获取选中的文本
  159.         # selectItem = self.listWidget.currentItem()  # 返回选中的项(仅限单选)
  160.         selectItem = self.listWidget.selectedItems()  # 返回一个包含item对象的list对象
  161.         select_item = [i.text() for i in selectItem]
  162.         # print(select_item)
  163.  
  164.         tb_cols = self.tableWidget.columnCount()
  165.         # 方法一：for循环
  166.         list_c = []
  167.         for i in select_item:
  168.             for c in range(tb_cols):
  169.                 if self.tableWidget.horizontalHeaderItem(c).text() == i:
  170.                     # print(c)
  171.                     list_c.append(c)
  172.         print(list_c)
  173.         # 方法二：for循环列表推导式
  174.         list_x = [c for i in select_item for c in range(tb_cols) if
  175.                   self.tableWidget.horizontalHeaderItem(c).text() == i]
  176.         print(list_x)
  177.  
  178.     def forLoopListDerivationFormula_dict(self):
  179.         # 获取选中的文本
  180.         # selectItem = self.listWidget.currentItem()  # 返回选中的项(仅限单选)
  181.         selectItem = self.listWidget.selectedItems()  # 返回一个包含item对象的list对象
  182.         select_item = [i.text() for i in selectItem]
  183.         # print(select_item)
  184.  
  185.         tb_cols = self.tableWidget.columnCount()
  186.         dict_x = {c: i for i in select_item for c in range(tb_cols) if
  187.                   self.tableWidget.horizontalHeaderItem(c).text() == i}
  188.         print(dict_x)
  189.  
  190.  
  191. if __name__ == '__main__':
  192.     app = QApplication(sys.argv)  # 在 QApplication 方法中使用，创建应用程序对象
  193.     myWin = MyMainWindow()  # 实例化 MyMainWindow 类，创建主窗口
  194.     myWin.show()  # 在桌面显示控件 myWin
  195.     sys.exit(app.exec_())  # 结束进程，退出程序

本文内容由网友自发贡献，转载请注明出处：https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/422058
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号