python - 将日期列添加到 python 中的 append 输出 csv 文件

Question

我使用下面的代码合并所有 csv 文件:每个文件下面有 10,000 行:

billing_report_2014-02-01.csvbilling_report_2014-02-02.csv:

fout=open("out.csv","a")
for num in range(1,10):
    print num
    for line in open("billing_report_2014-02-0"+str(num)+".csv"):
         fout.write(line) 
for num in range(10,29):
    print num
    for line in open("billing_report_2014-02-"+str(num)+".csv"):
         fout.write(line) 
fout.close()

但现在我想将新的日期列添加到 out.csv 文件，如何添加日期列并将值“2014-02-01”添加到我将 billing_report_2014-02-01 append 到 out.csv 的每一行， 和我将 billing_report_2014-02-02 append 到 out.csv 的每一行的“2014-02-02”值我该如何处理？

Answer 1

列出您要处理的文件名，然后从中获取数据，在输入文件上构建一个生成器以删除尾随的新行，并添加一个带有日期的新字段...例如:

filenames = [
  'billing_report_2014-02-01.csv',
  'billing_report_2014-02-02.csv'
]

with open('out.csv', 'w') as fout:
    for filename in filenames:
        to_append = filename.rpartition('_')[2].partition('.')[0]
        with open(filename) as fin:
            fout.writelines('{},{}\n'.format(line.rstrip(),to_append) for line in fin)