python - 从月份计算季度

Question

我有一个数据框，其中包含按月排列的多列

c1 c2 yyyy-01 yyyy-02 yyyy-03 yyyy-04 yyyy-05 yyyy-06 ...
 a  A     1      1       3      2       2        3     
 b  B     2      3       4      4       2        1

...大约有15年的数据我需要把它变成宿舍即

c1 c2 yyyy-q1 yyyy-q2 ...
 a  A    5      7
 b  B    9      7 
...

等等。有什么帮助吗？似乎 map 可以做到，但我不太精通 map 或 lambda。

Answer 1

1. 使用pd.to_datetime() 和to_period() 函数为每年的季度创建一个组变量；
2. 您可以将轴参数传递给 groupby() 函数以按列对数据帧进行分组，在本例中为列索引。

所以，你可以试试:

df.set_index(['c1', 'c2'], inplace=True)
df

df.groupby(pd.to_datetime(df.columns).to_period("Q"), axis=1).sum().reset_index()