python - 使用 DictWriter 读写 csv

Question

是否可以使用 DictWriter 读取和写入现有的 csv。如果没有，是否有可用的软件包可以执行以下操作:

csv:

event_type, flight_number, test_result
'ETA', '1234',''

脚本:

testData = csv.DictReader(open('test_data.csv', 'rU'))

for row in testData:

    eventType = row['event_type'].strip()
    flightNumber = row['flight_number'].strip()
    result = 6
    row['test_result'] = result

计算后，我想将结果的值写入“test_result”列的当前行。

Answer 1

经过多次修改，这种方法似乎是最好的，直到我找到更好的方法。

import csv, os

testData = csv.DictReader(open('test_data.csv', 'rU'))
rowHeaders = testData.fieldnames
resultFile = open('result_data.csv', 'w', newline='')
csvWriter = csv.DictWriter(resultFile, rowHeaders)
csvWriter.writeheader()
resultFile.close()

resultFile = open('result_data.csv', 'a', newline='')
csvWriter = csv.DictWriter(resultFile, rowHeaders)

for row in testData:

    eventType = row['event_type'].strip()
    flightNumber = row['flight_number'].strip()
    result = 7

    csvWriter.writerow({'event_type': eventType, 
                        'flight_number': flightNumber, 
                        'test_result': result})

它使用相同的精确数据+附加数据创建一个新文件

输出result_data.csv:

event_type, flight_number, test_result
ETA, 1234,7
ETD, 1034,7