将列添加到CSV文件并在该列中插入文件名

Question

我有一堆CSV文件，我将把它们合并到一个名为'Combined'的CSV文件中。对于每个CSV文件，一旦将数据附加到“组合”文件，我想在“组合”的第1列之前插入一个新列，并在该迭代中插入从中复制数据的CSV文件的名称。有什么办法可以在Python中做到这一点？

Answer 1

这可以按如下方式完成。首先打开一个CSV文件输出。现在使用Python的glob库来列出文件夹中的所有CSV文件。对于在CSV文件中的每一行，前缀的文件名作为第一列条目，然后将其写入output.csv：

import glob 
import csv 

with open('output.csv', 'w', newline='') as f_output: 
    csv_output = csv.writer(f_output) 

    for filename in glob.glob('*.csv'): 
     with open(filename, newline='') as f_input: 
      csv_input = csv.reader(f_input) 

      for row in csv_input: 
       row.insert(0, filename) 
       csv_output.writerow(row) 

因此，举例来说，如果你有这两个CSV文件中：

num.csv

1,2,3,4,5 
1,2,3,4,5 
1,2,3,4,5 

letter.csv

a,b,c,d,e,f 
a,b,c,d,e,f 
a,b,c,d,e,f 
a,b,c,d,e,f 

这将创建下列output.csv文件：

letter.csv,a,b,c,d,e,f 
letter.csv,a,b,c,d,e,f 
letter.csv,a,b,c,d,e,f 
letter.csv,a,b,c,d,e,f 
num.csv,1,2,3,4,5 
num.csv,1,2,3,4,5 
num.csv,1,2,3,4,5 

这里假定您使用的是Python 3.x.