Python Pandas 按功能分组

Question

我有这张 table

    uname   sid usage
0   Ahmad   a   5
1   Ahmad   a   7
2   Ahmad   a   10
3   Ahmad   b   2
4   Mohamad c   6
5   Mohamad c   7
6   Mohamad c   9

我想按 uname 和 side 分组，并且使用 column = group.max - group.min。但是，如果组计数为 1，则返回组 max

输出应该是

    uname   sid usage
0   Ahmad   a   5
1   Ahmad   b   2
2   Mohamad c   3    

Answer 1

首先，使用agg获取每组的min、max和size。
然后将 min 乘以 size > 1。如果是，它将等于 min，否则为 0。然后从 max 中减去它。

d1 = df.groupby(['uname', 'sid']).usage.agg(['min', 'max', 'size'])
d1['max'].sub(d1['min'].mul(d1['size'].gt(1))).reset_index(name='usage')