python - reading and writing csv with DictWriter -
is possible read , write existing csv dictwriter. if not, there package available following:
csv:
event_type, flight_number, test_result 'eta', '1234','' script:
testdata = csv.dictreader(open('test_data.csv', 'ru')) row in testdata: eventtype = row['event_type'].strip() flightnumber = row['flight_number'].strip() result = 6 row['test_result'] = result after calculation want write current row of 'test_result' column value of result.
after tinkering method seems best until find better way.
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) 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}) its creating new file same exact data + additional data
output result_data.csv :
event_type, flight_number, test_result eta, 1234,7 etd, 1034,7
Comments
Post a Comment