python - 根据列的唯一值分配增量值

Question

我正在尝试根据数据框中另一列的特定值向列添加增量值。这样...

col A   col B
 A        0
 B        1
 C        2
 A        3
 A        4
 B        5

会变成这样:

col A   col B
 A        1
 B        2
 C        3
 A        1
 A        1
 B        2
 C        3

已尝试使用 groupby 函数，但无法真正理解在 B 列上设置增量值。

有什么想法吗？

谢谢

Answer 1

我认为需要factorize :

df['col B'] = pd.factorize(df['col A'])[0] + 1
print (df)
  col A  col B
0     A      1
1     B      2
2     C      3
3     A      1
4     A      1
5     B      2

另一个解决方案:

df['col B'] = pd.Categorical(df['col A']).codes + 1
print (df)
  col A  col B
0     A      1
1     B      2
2     C      3
3     A      1
4     A      1
5     B      2