代码之家 › 专栏 › 技术社区 › HHH

如何在pandas中添加基于聚合函数的新列

group-by pandas

HHH · 技术社区 · 7 年前

我有一个熊猫数据框架,如下所示

A, B
----
a, 2
a, 5
a, 6
b, 1
b, 2

我想按栏目分组 A 并汇总列中的值 B 并将其附加为另一列并创建以下数据帧

A, B, SUM
--------
a, 2, 13
a, 5, 13
a, 6, 13
b, 1, 3
b, 2, 3

我在熊猫身上怎么做?

1 回复 | 直到 7 年前

BENY 7 年前

使用 transform

df.assign(SUM=df.groupby('A').B.transform('sum'))
Out[15]: 
   A  B  SUM
0  a  2   13
1  a  5   13
2  a  6   13
3  b  1    3
4  b  2    3

推荐文章

EvitaSchaap · 计算按另一列分组的标志百分比

1 年前

mxplk · 为具有条件的组设置布尔掩码

1 年前

Ryan Barnett · Pandas:向MultiIndex添加新列

1 年前

Leyla Elkhamlichi · 按特定值计数和分组

1 年前

KL_ · 熊猫分组和concat多行

1 年前

Nayr borcherds · Pandas Dataframe按组快速计数行数

1 年前

Nick · 如何在Python中使用Pandas将具有相同唯一ID的多行合并为一行[重复]

1 年前

DGMS89 · Groupby索引,并保持给定单个列的最大列值

1 年前

lubenthrust · 如何通过对pandas DataFrame的每一行进行分组来有效地选择顶部列?

1 年前

HizaCrenata · Pandas:在Groupby聚合后返回空白数组

1 年前