Question

我想在现有文件中添加新列。但是我添加的附加循环有点复杂。

输入文件：

testfile.csv

col1,col2,col3
1,2,3
3,4,5
4,6,7

输出我想要：

USA_testfile.csv

col1,col2,col3,country
1,2,3,USA
3,4,5,USA
4,6,7,USA

UK_testfile.csv

col1,col2,col3,country
1,2,3,UK
3,4,5,UK
4,6,7,UK

这就是我的尝试：

import csv
import sys
country_list= ['USA', 'UK']

def add_col(csv_file):
    for country in country_list:
        with open(csv_file, 'rb') as fin:
            with open(country+"_timeline_outfile_"+csv_file, 'wb') as fout:
                writer = csv.writer(fout, lineterminator='\n')
                reader = csv.reader(fin)

                all_rows =[]
                row = next(reader)
                row.append('country')
                all_rows.append(row)
                print all_rows

                for row in reader:
                    row.append(country)
                    all_rows.append(row)
                writer.writerows(all_rows)

add_col(sys.argv[1])

这是我得到的错误：

  File "write_to_csv.py", line 33, in add_col
    writer.writerows(all_rows)
ValueError: I/O operation on closed file

我试图关注此帖here

Answer 1

import csv

countries = ['USA', 'UK']
data = list(csv.reader(open('testfile.csv', 'rb')))

for country in countries:
    with open('{0}_testfile.csv'.format(country), 'wb') as f:
        writer = csv.writer(f)
        for i, row in enumerate(data):
            if i == 0:
                row = row + ['country']
            else:
                row = row + [country]
            writer.writerow(row)

Answer 2

我无法重现您的错误，但我清理了一下您的代码。没有理由为每种语言重新打开输入文件。

def add_col(csv_file):
    with open(csv_file, 'rb') as fin:
        reader = csv.reader(fin)
        for country in country_list:
            fin.seek(0) # jump to begin of file again
            with open(country+"_timeline_outfile_"+csv_file, 'wb') as fout:
                writer = csv.writer(fout, lineterminator='\n')
                header = next(reader)
                header.append('country')
                writer.writerow(header)
                for row in reader:
                    row.append(country)
                    writer.writerow(row)

python写入csv文件中的新列

2 个答案: