使用 DictWriter 更改列标题

Question

我正在处理一些代码，以从输入.txt文件生成两个新的 CSV 文件。.txt文件有 40 列数据，我正在编写两个新的 csv 文件，每个文件只接收.txt文件中的 2 列。

我的代码运行良好，几乎完全符合我的需求。最后一步是重命名两个新 csv 输出中的原始标头。

这是我的代码：

import csv

QATargetHeaders = ['FNAME','LNAME']
FinalTargetHeaders = ['PROJECT_CODE','PHONE_NUMBER']

with open('TEST.txt','r') as csv_input, open('Test_QA2.csv','w') as csv_output:
    reader = csv.DictReader(csv_input, delimiter='^')
    writer = csv.DictWriter(csv_output, fieldnames=QATargetHeaders, extrasaction='ignore')
    writer.writeheader()
    for line in reader:
        writer.writerow(line)
with open('TEST.txt','r') as csv_input, open('Test_FINAL.csv','w') as csv_output:
    reader = csv.DictReader(csv_input, delimiter='^')
    writer = csv.DictWriter(csv_output, fieldnames=FinalTargetHeaders, extrasaction='ignore')
    writer.writeheader()
    for line in reader:
        writer.writerow(line)

对于TEST_QA2.csv，我需要更改标题：

FNAME = First Name
LNAME = Last Name

对于TEST_FINAL.csv，我需要更改标题：

PROJECT_CODE = PROJECT ID
PHONE_NUMBER = DEVICE1

有人可以帮我弄清楚如何在这里重写标题吗？必须将 TargetHeader 定义为.txt文件中的确切列名，以便我的编写器知道要复制的内容。提前谢谢你

Answer 1

行是字典。字典键用作标头。因此，您必须在编写新的 csv 之前重命名密钥。您可以通过字典理解来做到这一点。

import csv

QATargetHeaders = ['First Name','Last Name']
FinalTargetHeaders = ['PROJECT ID','DEVICE1']

map_QATargetHeaders = (('FNAME', 'First Name'), ('LNAME', 'Last Name'))
map_FinalTargetHeaders = (('PROJECT_CODE', 'PROJECT ID'), ('PHONE_NUMBER', 'DEVICE1LNAME'))

with open('TEST.txt','r') as csv_input, open('Test_QA2.csv','w') as csv_output:
    reader = csv.DictReader(csv_input, delimiter='^')

    writer = csv.DictWriter(csv_output, fieldnames=QATargetHeaders, extrasaction='ignore')
    writer.writeheader()
    for line in reader:
        line = {map[1]:line[map[0]] for map in map_QATargetHeaders}
        writer.writerow(line)

with open('TEST.txt','r') as csv_input, open('Test_FINAL.csv','w') as csv_output:
    reader = csv.DictReader(csv_input, delimiter='^')

    writer = csv.DictWriter(csv_output, fieldnames=FinalTargetHeaders, extrasaction='ignore')
    writer.writeheader()
    for line in reader:
        line = {map[1]:line[map[0]] for map in map_FinalTargetHeaders}
        writer.writerow(line)

Answer 2

代替 writeheader，将 writerow 与字典一起使用，将旧标头映射到新标头。然后，您可以像以前一样写入其他行。

QATargetHeaders = {'FNAME': 'First Name','LNAME': 'Last Name'}
FinalTargetHeaders = {'PROJECT_CODE': 'PROJECT ID','PHONE_NUMBER': 'DEVICE1'}
...
    writer = csv.DictWriter(csv_output, fieldnames=QATargetHeaders, extrasaction='ignore')
    writer.writerow(QATargetHeaders)  # instead of writeheader
    for line in reader:
        writer.writerow(line)
...

（从 Python 3.7 开始，字典保留顺序，因此列将以正确的顺序显示。如果使用早期版本的 Python，则需要创建单独的字段名称列表，或使用 OrderedDict，以确保列以正确的顺序显示在 CSV 文件中。