python - 在数据帧的列上使用 sklearn 的 LabelEncoder

Question

如果我有一个数据框，请说 df，如果

df["levels"] = pd.Series(["low", "low", "med", "low", "med", "high"])

有没有办法将其更改为:

df["levels"] = pd.Series([0,0,1,0,1,2])

我尝试使用 preprocessing.LabelEncoder() 来转换它，但它只是折叠成 [0,1,2]。我知道我可以用 for 循环来做到这一点，但是如果已经有一些工具可以做到这一点，那就太好了任何帮助都是值得赞赏的!

Answer 1

有两种方法..op1 类别

pd.Series(["low", "low", "med", "low", "med", "high"]).astype('category').cat.codes
Out[1454]: 
0    1
1    1
2    2
3    1
4    2
5    0
dtype: int8

op2 分解

pd.factorize(pd.Series(["low", "low", "med", "low", "med", "high"]))[0]
Out[1455]: array([0, 0, 1, 0, 1, 2], dtype=int64)