Pandas_Map_column_with_values_in_dict.md

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