Map column with values in dict

​​
​

Template request | Bug report | Generate Data Product​
Tags: #pandas #dict #map #series
Author: Florent Ravenel​
Description: This notebook shows how to map a column of a Pandas DataFrame with values from a dictionary.
References:
https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.map.html
https://www.geeksforgeeks.org/python-pandas-dataframe-map/
Input
Import libraries
import pandas as pd
Setup Variables
data = {"Name": ["John", "Paul", "George", "Ringo"], "Age": [20, 21, 19, 18]}
mapping_dict = {20: "Young", 21: "Young", 19: "Adult", 18: "Adult"}
Model
Read dataframe
df = pd.DataFrame(data)
df
Map column with values in dict
df["Age_Group"] = df["Age"].map(mapping_dict)
Output
Display result
df

Merge Dataframes

Drop columns

Last modified 1mo ago