CSV文件未正确填充详细信息

时间:2018-10-02 04:39:10

标签: python csv export-to-csv

import csv
TextFileContent = open('tickets.txt')
with open('example4.csv', 'w') as csvfile:
    fieldnames = ['Author', 'ticket number', 'Revision']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
    writer.writeheader()
    for TextLine in TextFileContent:
        if 'Revision:' in TextLine:
            tmp=TextLine.replace('Revision:', "")
            print(tmp)
            writer.writerow({'Revision': tmp})
        elif 'Author:' in TextLine:
            tmp=TextLine.replace("Author:", "")
            print(tmp)
            writer.writerow({'Author': tmp})
        elif 'Contributes to:' in TextLine:
            tmp=TextLine.replace("Contributes to:", "")
            print(tmp)
            writer.writerow({'ticket number': tmp})

大家好,我已经在python脚本上面开发了从文本文件中提取“作者”,“票务”和“修订”详细信息,然后将该信息填充到CSV文件中。

现在,我可以提取所有信息,但数据未正确填充到CSV文件中。

文本文件内容如下所示

Revision: 22904
Author: Userx
Contributes to: CF-1159


Revision: 22887
Author: Usery
Contributes to: CF-955


Revision: 22884
Author: UserZ
Contributes to: CPL-7768

我想生成如下所示的CSV文件

Author  ticket number   Revision
Userx    CF-1159        22904
Usery    CF-955         22887
UserZ    CPL-7768       22884

1 个答案:

答案 0 :(得分:1)

您的代码在找到 any 字段后立即写一行,而不必等到读取完整的字段集。以下编辑将等待完整的设置,然后写入该行。

with open('/tmp/out.csv', 'w') as csvfile:
    fieldnames = ['Author', 'ticket number', 'Revision']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
    writer.writeheader()
    row = {}
    for TextLine in TextFileContent:
        if 'Revision:' in TextLine:
            row['Revision'] = TextLine.replace('Revision: ', "")
        elif 'Author:' in TextLine:
            row['Author'] = TextLine.replace("Author: ", "")
        elif 'Contributes to:' in TextLine:
            row['ticket number'] = TextLine.replace("Contributes to: ", "")
        if len(row) == len(fieldnames):
            writer.writerow(row)
            row = {}

请注意,除非所有记录都包含所有字段,否则这将无法正常工作。