如何使用Python在日期上拆分CSV文件

Question

我有一个csv文件，其中包含日期列，其格式为“ 1929-01-10”。 我想每年将这个巨大的文件拆分为多个单独的文件。 因此，对于date列中的每一年，都有一个单独的csv文件（最好是年份名称）。

我想用Python做到这一点

Answer 1

1. 获取我们必须在其中写入新文件的src位置。 和主要CSV文件名
2. 使用CSV模块读取和写入文件。
3. 使用collection defaultdict模块设置每个键值类型为list。
4. 读取器主文件，并从中迭代每一行。
5. 将每行的第一列除以-以获取年份值。
6. 使用年份值作为关键字，并在结果字典中追加行。
7. 现在我们将所有信息放入结果字典。
8. 迭代结果字典中的每个项目。
9. 再次使用CSV模块写入CSV文件。
10. 使用密钥作为文件名。

输入：main.csv

1929-01-10,1,a
1929-01-10,2,b
1930-01-10,3,c
1929-01-10,4,d
2015-01-10,5,e
2015-01-10,6,f
1929-01-10,7,g
2014-01-10,8,h

码：

src_path = "/home/vivek/Desktop/Work/stack/"
main_file = "/home/vivek/Desktop/Work/stack/main.csv"
import csv
import collections
import pprint

with open(main_file, "rb") as fp:
    root = csv.reader(fp, delimiter=',')
    result = collections.defaultdict(list)
    for row in root:
        year = row[0].split("-")[0]
        result[year].append(row)

print "Result:-"        
pprint.pprint(result)

for i,j in result.items():
    file_path = "%s%s.csv"%(src_path, i)
    with open(file_path, 'wb') as fp:
        writer = csv.writer(fp, delimiter=',')
        writer.writerows(j)

输出：

Result:-
defaultdict(<type 'list'>, {'2015': [['2015-01-10', '5', 'e'], ['2015-01-10', '6', 'f']], '1929': [['1929-01-10', '1', 'a'], ['1929-01-10', '2', 'b'], ['1929-01-10', '4', 'd'], ['1929-01-10', '7', 'g']], '1930': [['1930-01-10', '3', 'c']], '2014': [['2014-01-10', '8', 'h']]})

根据输入文件的一些结果文件：

1929.csv

1929-01-10,1,a
1929-01-10,2,b
1929-01-10,4,d
1929-01-10,7,g

2015.csv

2015-01-10,5,e
2015-01-10,6,f