基于字符串分类变量生成数字分类变量

Question

我正在尝试为字符串列表生成数字类别变量（类似于ID）。

详细来说，应该为每个“艺术家”分配一个ID。

Answer 1

这是您可以尝试的方式：

# create a mapping for artist to a number
maps = {ar:en for en, ar in enumerate(df['artist'].unique())}
df['artist_code'] = df['artist'].map(maps)

  artist  artist_code
0      a            0
1      a            0
2      a            0
3      b            1
4      c            2

样本数据

df = pd.DataFrame({'artist':['a','a','a','b','c']})

Answer 2

IIUC，使用Series.factorize

Series.factorize

或df['artist_code']=df['artis'].factorize()[0]

groupby.ngroup