Как панды конвертируют один столбец данных в другой?

Question

У меня есть кадр данных, сгенерированный пандами, следующим образом:

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

Answer 2

IIUC

s=df.CODE.eq('b').cumsum()
df['NUM']=df.CODE.where(df.CODE.eq('b'),s[~df.CODE.eq('b')].groupby(s).cumcount()+1)
df
Out[514]: 
    NO CODE NUM
0    1    a   1
1    2    a   2
2    3    a   3
3    4    a   4
4    5    a   5
5    6    a   6
6    7    b   b
7    8    b   b
8    9    a   1
9   10    a   2
10  11    a   3
11  12    a   4
12  13    b   b
13  14    a   1
14  15    a   2
15  16    a   3