python – pandas中的Groupby列后缀

发布时间：2020-12-20 12:03:41 所属栏目：Python 来源：网络整理

导读：给定一个数据帧： val1_aa val1_bb val2_aa val2_bb val2_cc val3_ccid 100 6 0 3 4 3 1200 0 1 0 0 1 0300 5 1 1 0 4 0400 0 3 1 5 7 1 我想按照每个列标题中的后缀顺序对所有列进行求和.我想要的输出： aa bb ccid 100 9 4 4200 0 1 1300 6 1 4400 1 8 8

给定一个数据帧：

val1_aa  val1_bb  val2_aa  val2_bb  val2_cc  val3_cc
id                                                       
100        6        0        3        4        3        1
200        0        1        0        0        1        0
300        5        1        1        0        4        0
400        0        3        1        5        7        1

我想按照每个列标题中的后缀顺序对所有列进行求和.我想要的输出：

aa  bb  cc
id             
100   9   4   4
200   0   1   1
300   6   1   4
400   1   8   8

我怎么得到这个？

答案发布在下面.

解决方法

您可以在轴= 1上使用lambda函数进行分组

In [4178]: df.groupby(lambda x: x.split('_')[-1],axis=1).sum()  # or x.split('_')[1]
Out[4178]:
     aa  bb  cc
id
100   9   4   4
200   0   1   1
300   6   1   4
400   1   8   8

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!