python - 按日期范围对数据框进行子集化

Question

尝试以有效的方式按日期范围分隔数据框，到目前为止我只提出:

mask0 = df['Date of survey'].between('2010-01-01', '2010-12-31')
mask1 = df['Date of survey'].between('2011-01-01', '2011-12-31')
mask2 = df['Date of survey'].between('2012-01-01', '2012-12-31')

...
maskn = df['Date of survey'].between('nnnn-01-01', 'nnnn-12-31')

任何想法将不胜感激!(我将使用掩码对数据帧进行子集化，并获取每年每个列变量的平均和)。

Answer 1

这里更好的是使用 DataFrame.resample使用 mean 和 sum 等聚合函数按年计算:

df1 = df.resample('A', on='Date of survey').agg(['mean','sum'])

或者使用DataFrame.groupby按年 Series.dt.year :

df2 = df.groupby(df['Date of survey'].dt.year).agg(['mean','sum'])