重命名 Pandas aggreate 函数中的标头



我正在尝试使用聚合将多个函数应用于我的数据帧,

事实证明,它将添加另一行来指示函数的名称,

我想知道是否有任何方法可以提前重命名标题,这样我就不会有多余的行

feature_setup = {'A':['min','max','mean','std','sum'] , 
'B': ['min','max','mean','std','sum'], 
'C': ['sum'],
'D': ['sum']}
features = ['A','B,'C','D']
dataframe = df[features].rolling(w_size).agg(feature_setup)

我的结果数据帧就像

A    B    C    D   A    B    A     B   A    B
0  min  max  sum  sum  mean mean  std  std  sum sum
1   0    0    0    0    0    0      0   0    0    0
2  ....

有没有办法提前重命名标题,或合并第 0 行和标题?

谢谢!

将多索引列连接成基于字符串的列,例如

x = df[features].rolling(2).agg(feature_setup)
x.columns = ['_'.join(col) for col in x]

列将是

A_min   A_max   A_mean  A_std   A_sum   B_min   B_max   B_mean  B_std   B_sum   C_sum

最新更新