В сохраненном файле отсутствуют заголовки столбцов - PullRequest
Новая
       51

В сохраненном файле отсутствуют заголовки столбцов

2 голосов
/ 06 мая 2020

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

def download_results(self):
    try:
        path = QFileDialog.getSaveFileName(MainWindow, 'Save results', os.getenv('HOME'), 'CSV(*.csv)')
        if path[0] != '':
            with open(path[0], 'w') as csv_file:
                writer = csv.writer (csv_file, dialect = 'excel', delimiter = ',')
                for row in range(self.analysis_table.rowCount()):
                    row_data = []
                    for column in range(self.analysis_table.columnCount()):
                        item = self.analysis_table.item(row, column)
                        if item is not None:
                            row_data.append(item.text())
                        else:
                            row_data.append('')
                    writer.writerow(row_data)
        QMessageBox.information(MainWindow,"Success","Succeeded")
    except:
        QMessageBox.warning(MainWindow,"Failed","Operation failed.")

Ответы [ 2 ]

2 голосов
/ 06 мая 2020

Вы должны извлечь текст из QTableWidgetItem, используя метод horizontalHeaderItem(): 

def download_results(self):
    try:
        path, _ = QFileDialog.getSaveFileName(
            MainWindow, "Save results", os.getenv("HOME"), "CSV(*.csv)"
        )
        if path:
            with open(path, "w", <b>newline=''</b>) as csv_file:
                writer = csv.writer(csv_file)
                <b>headers = []
                for c in range(self.analysis_table.columnCount()):
                    it = self.analysis_table.horizontalHeaderItem(c)
                    if it is not None:
                        headers.append(it.text())
                    else:
                        headers.append("")
                writer.writerow(headers)</b>
                for row in range(self.analysis_table.rowCount()):
                    row_data = []
                    for column in range(self.analysis_table.columnCount()):
                        item = self.analysis_table.item(row, column)
                        if item is not None:
                            row_data.append(item.text())
                        else:
                            row_data.append("")
                    writer.writerow(row_data)
        QMessageBox.information(MainWindow, "Success", "Succeeded")
    except:
        QMessageBox.warning(MainWindow, "Failed", "Operation failed.")
0 голосов
/ 06 мая 2020

Это сделало это. Я благодарен участникам 

def download_results(self):
    try:
        path = QFileDialog.getSaveFileName(MainWindow, "Save results", os.getenv("HOME"), "CSV(*.csv)")
        if path != "":
            with open(path[0], "w", newline='') as csv_file:
                writer = csv.writer(csv_file)
                headers = []
                for c in range(self.analysis_table.columnCount()):
                    it = self.analysis_table.horizontalHeaderItem(c)
                    if it is not None:
                        headers.append(it.text())
                    else:
                        headers.append("")
                writer.writerow(headers)
                for row in range(self.analysis_table.rowCount()):
                    row_data = []
                    for column in range(self.analysis_table.columnCount()):
                        item = self.analysis_table.item(row, column)
                        if item is not None:
                            row_data.append(item.text())
                        else:
                            row_data.append("")
                    writer.writerow(row_data)
        QMessageBox.information(MainWindow, "Success", "Succeeded")
    except:
        QMessageBox.warning(MainWindow, "Failed", "Operation failed.")
...