python - 通过 Pandas 操作访问特定组

Question

    A   B0   1  121   1  132   1  153   2  164   2  195   2  206   3  327   3  298   3  259   4   310  4   511  4   7

我有一个 pandas 数据框，如上所示。我想按 1 2 3 4 等值对“A”列进行分组，然后从第 4 组中减去所有组的 B 列的值。我该怎么做？

我的结果应该是这样的:

    A   B   C1   12  91   13  81   15  82   16  132   19  142   20  133   32  293   29  243   25  184   3   04   5   04   7   0

Answer 1

您可以使用groupby 和apply。要获取 A=4 的组，请调用 GroupBy.get_group(4)。

g = df.groupby('A')
df['C'] = g.apply(lambda x: x['B'] - g.get_group(4)['B'].values).values
df

    A   B   C
0   1  12   9
1   1  13   8
2   1  15   8
3   2  16  13
4   2  19  14
5   2  20  13
6   3  32  29
7   3  29  24
8   3  25  18
9   4   3   0
10  4   5   0
11  4   7   0