python - pandas 如何将一列数据转换为另一列数据？

Question

我有一个pandas生成的dataframe，如下:

NO  CODE
1   a
2   a
3   a
4   a
5   a
6   a
7   b
8   b
9   a
10  a
11  a
12  a
13  b
14  a
15  a
16  a

我想将 CODE 列数据转换为 NUM 列。编码规则如下:

NO CODE NUM
1   a   1
2   a   2
3   a   3
4   a   4
5   a   5
6   a   6
7   b   b
8   b   b
9   a   1
10  a   2
11  a   3
12  a   4
13  b   b
14  a   1
15  a   2
16  a   3

谢谢!

Answer 1

尝试:

a_group = df.CODE.eq('a')

df['NUM'] = np.where(a_group, 
                     df.groupby(a_group.ne(a_group.shift()).cumsum())
                       .CODE.cumcount()+1, 
                     df.CODE)

在

df = pd.DataFrame({'CODE':list('baaaaaabbaaaabbaa')})

产量

    CODE    NUM
--  ------  -----
 0  b       b
 1  a       1
 2  a       2
 3  a       3
 4  a       4
 5  a       5
 6  a       6
 7  b       b
 8  b       b
 9  a       1
10  a       2
11  a       3
12  a       4
13  b       b
14  b       b
15  a       1
16  a       2