按月重塑 Pandas 数据框

Question

任务是改造下面table

import pandas as pd
import numpy as np

index = pd.date_range('2000-1-1', periods=700, freq='D')
df = pd.DataFrame(np.random.randn(700), index=index, columns=["values"])

df.groupby(by=[df.index.year, df.index.month]).sum()

In[1]: df

Out[1]: 
           values
2000 1   1.181000
     2  -8.005783
     3   6.590623
     4  -6.266232
     5   1.266315
     6   0.384050
     7  -1.418357
     8  -3.132253
     9   0.005496
     10 -6.646101
     11  9.616482
     12  3.960872
2001 1  -0.989869
     2  -2.845278
     3  -1.518746
     4   2.984735
     5  -2.616795
     6   8.360319
     7   5.659576
     8   0.279863
     9  -5.220678
     10  5.077400
     11  1.332519

看起来像这样

      Jan  Feb  Mar  Apr  May  Jun  Jul  Aug  Sep  Oct  Nov  Dec  
2000  1.2  -8.0 6.6  -6.3 1.2  0.4  -1.4 -3.1 0.0  -6.6 9.6  3.9    
2001  -0.9 -2.8 -1.5 3.0  -2.6 8.3  5.7  0.3  -5.2 5.1  1.3

此外，我需要添加一个额外的列来汇总年度值，如下所示

      Jan  Feb  Mar  Apr  May  Jun  Jul  Aug  Sep  Oct  Nov  Dec  Year
2000  1.2  -8.0 6.6  -6.3 1.2  0.4  -1.4 -3.1 0.0  -6.6 9.6  3.9  4.7
2001  -0.9 -2.8 -1.5 3.0  -2.6 8.3  5.7  0.3  -5.2 5.1  1.3       10.7

有没有快速 pandas 解决这个问题的关键方法？

Answer 1

额外的 Year 专栏，您可以通过

df['Year'] = df.sum(axis=1)

它将按行对数据帧求和（由于 axis=1），并将其存储在新列中。

Answer 2

你可以这样做：

import pandas as pd
import numpy as np

index = pd.date_range('2000-1-1', periods=700, freq='D')
df = pd.DataFrame(np.random.randn(700), index=index, columns=["values"])

l = [df.index.strftime("%Y"), df.index.strftime("%b"), df.index.strftime("%d")]
df.index = l
df=df.groupby(level=[-3,-2]).sum().unstack(-1)
df['Year'] = df.sum(axis=1)
df

输出：

Answer 3

在 groupby

中使用 strftime('%b')

df['values'].groupby([df.index.year, df.index.strftime('%b')]).sum().unstack()

保留月份顺序

df['values'].groupby([df.index.year, df.index.strftime('%b')], sort=False).sum().unstack()

最后 'Year'

df['values'].groupby([df.index.year, df.index.strftime('%b')], sort=False).sum() \
    .unstack().assign(Year=df.groupby(df.index.year).sum())

Answer 4

唯一的变化是您需要 unstack DF 将其转换为宽格式。获得整数月份数字后，您可以通过指定 %m 指令作为要考虑的格式，将它们转换为 datetime。获得它之后，通过 strftime.

的帮助，使用它来检索它的字符串表示形式

通过指定 axis=1.

计算各列的总和来计算年份

np.random.seed(314)
fr = df.groupby([df.index.year, df.index.month]).sum().unstack(fill_value=0)
fr.columns = pd.to_datetime(fr.columns.droplevel(0), format='%m').strftime('%b')
fr['Year'] = fr.sum(1)

按月重塑 Pandas 数据框

Reshaping Pandas dataframe by months

python

pivot

sum

time-series

pandas