Как применить LabelEncoder для определенного столбца в кадре данных Pandas

Question

У меня есть набор данных, загруженный фреймом данных, где метка класса должна быть закодирована с использованием LabelEncoder из scikit-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 в кадре данных закодированными метками.