将熊猫中的字符串单词转换为numpy整数

Question

我正在阅读熊猫csv文件：

  class
0 white
1 green
2 white

如何获取numpy数组，该数组将类名称更改为整数：

[0,1,0]

Answer 1

您可以使用categorical column as an index（自this起）：

df2 = pd.DataFrame(df.temp)
df2.index = pd.CategoricalIndex(df.cc)