python pandas dataframe 列 转换为离散值

python pandas dataframe 列 转换为离散值

import pandas as pd
import numpy as np

df1 = pd.DataFrame(
    {
        '日期': ['2021-7-2', '2021-8-2', '2021-9-2', '2021-10-2'],
        'A': [1, 2, 3, 4],
        'B': [5, 6, 7, 8],
        'C': [9, 10, 11, 12],
        'D': [13, 14, 15, 16],
        'E': ['年龄', '性别', '职业', '姓名'],
    },
    index=[0, 1, 2, 3])
print(df1)

         日期  A  B   C   D   E
0   2021-7-2  1  5   9  13  年龄
1   2021-8-2  2  6  10  14  性别
2   2021-9-2  3  7  11  15  职业
3  2021-10-2  4  8  12  16  姓名

df1['E'] = pd.Categorical(df1['E'])  # 将 E 列转换为离散数值
df1['E'] = df1.E.cat.codes  # 将 E 列转换为离散数值
print(df1)

         日期  A  B   C   D  E
0   2021-7-2  1  5   9  13  1
1   2021-8-2  2  6  10  14  2
2   2021-9-2  3  7  11  15  3
3  2021-10-2  4  8  12  16  0