合并多个 csv 文件并添加一个新列

Question

我有一堆 csv 文件，我需要合并到一个文件中，但有一个额外的日期列

xxxxx20150216.csv

xxxxx20130802.csv

xxxxx20130803.csv

xxxxx20130804.csv

我正在使用 (http://cbrownley.wordpress.com/2014/03/09/pythons-voracious-glob-module/) 中的以下代码来合并它们

import csv
import glob
import os
import sys

data_path = ""
outfile_path = "alldata.csv"
filewriter = csv.writer(open(outfile_path,'wb'))
file_counter = 0
for input_file in glob.glob(os.path.join(data_path,'*.csv')):
        with open(input_file,'rU') as csv_file:
                filereader = csv.reader(csv_file)
                if file_counter < 1:
                        for row in filereader:
                                filewriter.writerow(row)
                else:
                        header = next(filereader,None)
                        for row in filereader:
                                filewriter.writerow(row)
        file_counter += 1

但现在我需要从文件名中提取日期并将其与其他行一起添加为列。实现此目的最简单的方法是什么？

Answer 1

那……呢：

    with open(input_file,'rU') as csv_file:
        filereader = csv.reader(csv_file)
        name, ext = os.path.splitext(input_file)
        date = name[-8:]
        if file_counter < 1:
            for i, row in enumerate(filereader):
                if i==0: row.append('Date')
                else: row.append(date)
                filewriter.writerow(row)
            else:
                header = next(filereader,None)
                for row in filereader:
                    row.append(date)
                    filewriter.writerow(row)

唯一棘手的部分是从第一个 CSV 文件中获取标题！-)