带有cumsum的熊猫中的GroupBy类[重复]

Question

一个快速的，但我有点坚持这一点。我有一个包含 3 个类的数据框：0、1、2。这个想法是使用可能的 groupby 来计算每个类的 cumsum，尽管可能还有其他方法。

这是我的 df：

| classes |
|    1    |
|    0    |
|    1    |
|    2    |
|    1    |
|    2    |
|    0    |
|    0    |

我想查看的“ID”列：

| classes | ID |
|    1    | 1  |
|    0    | 1  |
|    1    | 2  |
|    2    | 1  |
|    1    | 3  |
|    2    | 2  |
|    0    | 2  |
|    0    | 3  |
....etc...

有什么想法吗？

Answer 1

你需要在这里使用 cumcount 而不是 cumsum :)

df['Id'] = (df.groupby('classes').cumcount() +1)

Answer 2

您正在寻找GroupBy.cumcount 我相信：

df["ID"] = df.groupby("classes").cumcount() + 1

得到

    classes   ID
0          1   1
1          0   1
2          1   2
3          2   1
4          1   3
5          2   2
6          0   2
7          0   3