使用熊猫根据结果分组映射值[重复]

Question

如何改变这个数据框

Index_Value         Value
value_1             1
value_1             2
value_1             3
value_2             33
value_3             111
value_3             222

到

Index_Value         Value       Counter
value_1             1           1
value_1             2           2
value_1             3           3
value_2             33          1
value_3             111         1
value_3             222         2

使用熊猫。我开始使用 for 循环来实现这一点，但对于大型数据集，性能可能会很差。

Answer 1

你应该在groupby之后使用cumcount：

import pandas as pd

df = pd.DataFrame({"Index_Value": ["v1", "v1", "v1", "v2", "v3", "v3"], "Value": list(range(6))})
df.set_index("Index_Value", inplace=True)

df.assign(Counter = df.groupby(df.index).cumcount() + 1)

输出

             Value  Counter
Index_Value
v1               0        1
v1               1        2
v1               2        3
v2               3        1
v3               4        1
v3               5        2