LabelEncoding到pandas中的多个列

Posted 2021-04-11

我目前正在研究泰坦尼克号数据集。它由4-5个非数字列组成。我想应用sklearn.LabelEncoder类来获取这些非数字列的编码值。毫无疑问，我可以逐个将这种方法应用于每一列。但是，当有超过20-30个这样的专栏时，这项工作将变得更加乏味。由于我知道这些非数字列的名称，有没有任何复杂的方法以轻松的方式这样做？

答案

只需在选择对象类型后运行循环

obj_cols = df.select_dtypes(include=[object])

for i in obj_cols:
    df[i+'label'] = le.fit_transform(df[i])