如何将LabelEncoder应用于Pandas数据帧中的特定列

Question

我有一个由dataframe加载的数据集,其中类标签需要使用LabelEncoderscikit-learn 进行编码.该列label是类标签列,它具有以下类:

[‘Standing’, ‘Walking’, ‘Running’, ‘null’]

为了执行标签编码,我尝试了以下但它不起作用.我该如何解决？

from sklearn import preprocessing
import pandas as pd

df = pd.read_csv('dataset.csv', sep=',') 
df.apply(preprocessing.LabelEncoder().fit_transform(df['label']))

Answer 1

你可以尝试如下:

le = preprocessing.LabelEncoder()
df['label'] = le.fit_transform(df.label.values)

或者以下也会起作用:

df['label'] = le.fit_transform(df['label'])

它将label用编码标签替换数据框中的原始值.