按日期拆分 CSV

Question

我希望有一个相对简单的解决方案来解决我的问题：

我有一个包含一系列数据点的 csv，但它们都包含一个日期字段。

我希望能够根据日期字段的月份将 csv 拆分为多个文件。

例如：我希望能够将 2015 年 3 月之前的所有记录放在一个文件中，将 2015 年 4 月之前的所有记录放在另一个文件中，直到 2016 年 10 月之前的所有记录，等等。

在这种情况下，文件之间会有很多重复记录。

有没有办法用一些简单的python代码来做到这一点，或者有没有更简单的方法？

提前致谢

Answer 1

此代码假定日期字段位于第一列并标记为“dates”。我们使用 pandas 将数据读入数据框并将['dates'] 作为列传递以转换为日期对象。然后，我们使用年份和月份获取数据框的不同切片来创建子集视图。然后将每个视图转储到格式为 year_month.csv 的新 csv

import pandas as pd

df = pd.read_csv('filename.csv', parse_dates=['dates'])

for year in df.dates.apply(lambda x: x.year).unique():
    for month in df.dates.apply(lambda x: x.month).unique():
        view = df[df.dates.apply(lambda x: x.month == month and x.year==year)]
        if view.size:
            view.to_csv('{}_{:0>2}.csv'.format(year, month))

可能有更好的方法来做到这一点，但这将完成工作。