如何在熊猫中将每月数据转换为季度

时间:2016-11-08 22:03:00

标签: python pandas dataframe group-by

我有月度数据。我希望将它转换为1月份q1开始的3个月的“期间”。因此,在下面的示例中,前3个月的聚合将转换为q2的开始(所需格式:1996q2)。由3个月值组合而成的数据值是3列的平均值(平均值)。从概念上讲,并不复杂。有谁知道如何一举做到这一点?潜在地,我可以通过循环来做很多艰苦的工作,只是硬编码它,但我是熊猫新手,寻找比蛮力更聪明的东西。

1996-04   1996-05 1996-06  1996-07 .....
25          19       37      40

所以我在寻找:

1996q2  1996q3   1996q4  1997q1  1997q2 .....
 avg      avg      avg     ...     ...

1 个答案:

答案 0 :(得分:26)

您可以将pd.PeriodIndex(..., freq='Q')groupby(..., axis=1)结合使用:

In [63]: df
Out[63]:
   1996-04  1996-05  2000-07  2000-08  2010-10  2010-11  2010-12
0        1        2        3        4        1        1        1
1       25       19       37       40        1        2        3
2       10       20       30       40        4        4        5

In [64]: df.groupby(pd.PeriodIndex(df.columns, freq='Q'), axis=1).mean()
Out[64]:
   1996Q2  2000Q3    2010Q4
0     1.5     3.5  1.000000
1    22.0    38.5  2.000000
2    15.0    35.0  4.333333

UPDATE :将结果DF中的列作为period dtype的字符串整数获取:

In [66]: res = (df.groupby(pd.PeriodIndex(df.columns, freq='Q'), axis=1)
                  .mean()
                  .rename(columns=lambda c: str(c).lower()))

In [67]: res
Out[67]:
   1996q2  2000q3    2010q4
0     1.5     3.5  1.000000
1    22.0    38.5  2.000000
2    15.0    35.0  4.333333

In [68]: res.columns.dtype
Out[68]: dtype('O')