Как заменить значения столбцов в pandas на идентификаторы?

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14

import pandas as pd

# Создание DataFrame
data = {'Страна': ['Россия', 'США', 'Китай', 'Германия'],
        'Континент': ['Европа', 'Америка', 'Азия', 'Европа']}
df = pd.DataFrame(data)

# Создание словаря с соответствием идентификаторов
идентификаторы = {i: j for i, j in enumerate(df['Страна'].unique())}

# Замена значений столбца на идентификаторы
df['Страна'] = df['Страна'].replace(идентификаторы)

print(df)

1
2
3
4
5

   Страна Континент
0       0    Европа
1       1   Америка
2       2      Азия
3       3    Европа