当 pandas 数据聚合在 python 时，如何计算每组尾部的总和|均值|中位数

Question

我有类似 following.which 的数据采用 pandas 数据帧格式。

A  B  C  D  E  F  G
1  1  2  3  1  4  2
1  1  2  4  5  6  7
1  1  2  3  2  3  2
1  1  2  4  5  6  7
2  1  2  3  2  3  4
2  1  2  3  4  3  3
2  1  2  4  5  6  7

此处agg_lvl=['A','B','C']

当数据聚合到 agg_lvl.

时，我想通过使用每个组中的 tail(2) 记录来计算 G 变量的均值|中值|总和

我的预期输出是这样的：

均值的预期输出：

A  B  C  G
1  1  2  4.5
2  1  2   5

中位数和总和的输出也是相同的，但是代替均值我们必须考虑中位数和总和值。

为此，我尝试了以下代码，但没有得到预期的输出。

df.groupby(agg_lvl,as_index=False).tail(2).agg({'G':'mean'})

谁能帮我解决这个问题。

提前致谢。

Answer 1

使用 GroupBy.transform 而不是 agg 用于 return 与 tail:

过滤后的 DataFrame 具有相同形状的新列

agg_lvl=['A','B','C']
df = df.groupby(agg_lvl,as_index=False).tail(2)
df['G'] = df.groupby(agg_lvl)['G'].transform('mean')
print (df)
   A  B  C  D  E  F    G
2  1  1  2  3  2  3  4.5
3  1  1  2  4  5  6  4.5
5  2  1  2  3  4  3  5.0
6  2  1  2  4  5  6  5.0

编辑：

df = df.groupby(agg_lvl,as_index=False).tail(2).groupby(agg_lvl,as_index=False)['G'].mean()
print (df)
   A  B  C    G
0  1  1  2  4.5
1  2  1  2  5.0

当 pandas 数据聚合在 python 时，如何计算每组尾部的总和|均值|中位数

how to calculate sum|mean|median for tail of each group when pandas data aggregated in python

python

tail

pandas

pandas-groupby