Ordinal Encoding

SKlearn - Ordinal Encoding

import pandas as pd
from sklearn.preprocessing import OrdinalEncoder

# Example DataFrame with job hierarchy
data = {
    "Name": ["Alice", "Bob", "Charlie", "Diana", "Eve"],
    "Job_Level": ["Junior", "Intern", "Manager", "Senior", "Director"],
}

df = pd.DataFrame(data)
print("Original DataFrame:")
print(df)

# Define the hierarchy order
job_hierarchy = ["Intern", "Junior", "Senior", "Manager", "Director"]

# Initialize OrdinalEncoder with the specified order
encoder = OrdinalEncoder(categories=[job_hierarchy])

# Fit and transform the 'Job_Level' column
df["Job_Level_encoded"] = encoder.fit_transform(df[["Job_Level"]])

print("\nDataFrame after encoding 'Job_Level' column:")
print(df)