Ordinal Encoder

PHOTO EMBED

Mon Nov 18 2024 15:52:32 GMT+0000 (Coordinated Universal Time)

Saved by @login123

from sklearn.preprocessing import OrdinalEncoder

data = {'Size': ['Small', 'Medium', 'Large', 'Medium']}
df = pd.DataFrame(data)

encoder = OrdinalEncoder(categories=[['Small', 'Medium', 'Large']])

df['Size_encoded'] = encoder.fit_transform(df[['Size']])
print(df)
content_copyCOPY