正在努力从QTableWidget导出csv数据[duplicate]

Question

因此，我试图编写一个代码，将csv文件加载到每个选项卡上的QTableWidget中，然后可以在其顶部编辑和保存。然而，当我保存时，程序由于各种原因崩溃。虽然没有设置自动加载，但我试图一次解决一个问题！

我看到过很多帖子似乎都在问同样的问题，但在实现它们时，它们似乎不起作用，有些是因为它们使用tableview而不是QTableWidget，有些可能是因为我没有正确地实现它，因为我还不熟悉编码。

类MyTabs（QWidget）：

def __init__(self, parent):
    super(QWidget, self).__init__(parent)
    layout = QVBoxLayout(self)

    # Initialize tab screen
    self.tabs = QTabWidget()
    self.tabmon = QWidget()
    self.tabtue = QWidget()

    # Add tabs
    self.tabs.addTab(self.tabmon, "Monday")
    self.tabs.addTab(self.tabtue, "Tuesday")

    #Save Button

    self.buttonSavemon = QtWidgets.QPushButton('Save', self)
    self.buttonSavemon.clicked.connect(self.handleSavemon)

    self.buttonSavetue = QtWidgets.QPushButton('Save', self)
    self.buttonSavetue.clicked.connect(self.handleSavetue)

    #Initiate Tables
    self.createTable()

    # Create Monday tab
    self.tabmon_layout = QVBoxLayout(self.tabmon)
    self.tabmon_layout.addWidget(self.tablewidgetmon)
    self.tabmon_layout.addWidget(self.buttonSavemon)

    # Create Tuesday tab
    self.tabtue_layout = QVBoxLayout(self.tabtue)
    self.tabtue_layout.addWidget(self.tablewidgettue)
    self.tabtue_layout.addWidget(self.buttonSavetue)

    # Add tabs to widget
    layout.addWidget(self.tabs)

def createTable(self):
    #Monday Table
    self.tablewidgetmon = QTableWidget()
    self.tablewidgetmon.setRowCount(10)
    self.tablewidgetmon.setColumnCount(2)
    self.tablewidgetmon.setHorizontalHeaderLabels(["Time", "File Name"])

    #Tuesday Table
    self.tablewidgettue = QTableWidget()
    self.tablewidgettue.setRowCount(12)
    self.tablewidgettue.setColumnCount(2)
    self.tablewidgettue.setHorizontalHeaderLabels(["Time", "File Name"])

def handleSavemon(self):
    with open('monschedule.csv', 'wb') as stream:
        writer = csv.writer(stream)
        for row in range(self.tablewidgetmon.rowCount()):
            rowdata = []
            for column in range(self.tablewidgetmon.columnCount()):
                item = self.tablewidgetmon.item(row, column)
                if item is not None:
                    rowdata.append(
                        unicode(item.text()).encode('utf8'))
                else:
                    rowdata.append('')
            writer.writerow(rowdata)

def handleSavetue(self):
    with open('tueschedule.csv', "w") as fileOutput:
        writer = csv.writer(fileOutput)
        for rowNumber in range(self.tablewidgettue.rowCount()):
            fields = [
                self.tablewidgettue.data(
                    self.tablewidgettue.index(rowNumber, columnNumber),
                    QtCore.Qt.DisplayRole
                )
                for columnNumber in range(self.tablewidgettue.columnCount())
            ]
            writer.writerow(fields)

我尝试了两种方法来保存这里，周一函数似乎是最有希望的，但是给我一个错误，因为数据是字符串。

从查看csv文件后，它似乎已清除原始文件，但没有保存新数据。

Answer 1

试试吧：

from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore    import *
from PyQt5.QtGui     import *
from PyQt5.QtWidgets import *
import csv


class MyTabs(QWidget):
    def __init__(self, parent=None):
        super(QWidget, self).__init__(parent)
        layout = QVBoxLayout(self)

        # Initialize tab screen
        self.tabs = QTabWidget()
        self.tabmon = QWidget()
        self.tabtue = QWidget()

        # Add tabs
        self.tabs.addTab(self.tabmon, "Monday")
        self.tabs.addTab(self.tabtue, "Tuesday")

        #Save Button

        self.buttonSavemon = QtWidgets.QPushButton('Save', self)
        self.buttonSavemon.clicked.connect(self.handleSavemon)

        self.buttonSavetue = QtWidgets.QPushButton('Save', self)
        self.buttonSavetue.clicked.connect(self.handleSavetue)

        #Initiate Tables
        self.createTable()

        # Create Monday tab
        self.tabmon_layout = QVBoxLayout(self.tabmon)
        self.tabmon_layout.addWidget(self.tablewidgetmon)
        self.tabmon_layout.addWidget(self.buttonSavemon)

        # Create Tuesday tab
        self.tabtue_layout = QVBoxLayout(self.tabtue)
        self.tabtue_layout.addWidget(self.tablewidgettue)
        self.tabtue_layout.addWidget(self.buttonSavetue)

        # Add tabs to widget
        layout.addWidget(self.tabs)

    def createTable(self):
        #Monday Table
        self.tablewidgetmon = QTableWidget()
        self.tablewidgetmon.setRowCount(10)
        self.tablewidgetmon.setColumnCount(2)
        self.tablewidgetmon.setHorizontalHeaderLabels(["Time", "File Name"])

        #Tuesday Table
        self.tablewidgettue = QTableWidget()
        self.tablewidgettue.setRowCount(12)
        self.tablewidgettue.setColumnCount(2)
        self.tablewidgettue.setHorizontalHeaderLabels(["Time", "File Name"])


    def handleSavemon(self):
#        with open('monschedule.csv', 'wb') as stream:
        with open('monschedule.csv', 'w') as stream:                  # 'w'
            writer = csv.writer(stream, lineterminator='\n')          # + , lineterminator='\n'
            for row in range(self.tablewidgetmon.rowCount()):
                rowdata = []
                for column in range(self.tablewidgetmon.columnCount()):
                    item = self.tablewidgetmon.item(row, column)
                    if item is not None:
#                        rowdata.append(unicode(item.text()).encode('utf8'))
                        rowdata.append(item.text())                   # +
                    else:
                        rowdata.append('')

                writer.writerow(rowdata)


    def handleSavetue(self):
        with open('tueschedule.csv', "w") as fileOutput:
            writer = csv.writer(fileOutput)
            for rowNumber in range(self.tablewidgettue.rowCount()):
# +                
                fields = [
                    self.tablewidgettue.item(rowNumber, columnNumber).text() \
                            if self.tablewidgettue.item(rowNumber, columnNumber) is not None else ""
                    for columnNumber in range(self.tablewidgetmon.columnCount())
                ]                

                writer.writerow(fields)


if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    main = MyTabs()
    main.show()
    sys.exit(app.exec_()) 

Answer 2

您将以二进制文件而不是文本文件的形式打开文件。这就是为什么函数需要一个类似字节的对象。

要解决此问题，请删除open调用中的“b”标志：

with open('monschedule.csv', 'w') as stream:

顺便说一下，行rowdata。追加（unicode（item.text（））.encode（'utf8'））应该是rowdata。追加（item.text（））