基于相同值python数据框设置索引

Question

在python数据框中，我有一个这样的数据框

index	column A
0	a
1	a
2	b
3	c
4	c
5	c
6	c

我想创建一个列，该列将根据同一列的值设置索引

index	column A	setIndex
0	a	0
1	a	1
2	b	0
3	c	0
4	c	1
5	c	2
6	c	3

Answer 1

你可以使用.groupby() + .cumcount()，如下：

df['setIndex'] = df.groupby('column A').cumcount()

结果：

print(df)

  column A  setIndex
0        a         0
1        a         1
2        b         0
3        c         0
4        c         1
5        c         2
6        c         3