One-Hot Encoding
Easy
feature-engineering
Stripe
LinkedIn
Implement one-hot encoding for a categorical column. Given a list of values, return a list of dictionaries where each dictionary maps category names to 0 or 1.
Example
one_hot_encode(["red", "blue", "red", "green"])
# => [
# {"red": 1, "blue": 0, "green": 0},
# {"red": 0, "blue": 1, "green": 0},
# {"red": 1, "blue": 0, "green": 0},
# {"red": 0, "blue": 0, "green": 1}
# ]
Test Cases
Python Editor
Output
Click "Run" to see results...