CODEWITHSUNDEEP
pandasdictionaryseries
Dan_Lee
Dan_Lee

Reputation: 97

Pandas replace values using dictionary with a list of values

I have the following dictionary and Series.

product_type_dic = {'Facial Care':['Facial_cleanser', 'Beard_oil'], 'Eye Care':['Around-eye_cream', 'Eyeliner']}

s = pd.Series(['Facial_cleanser', 'Beard_oil', 'Around-eye_cream', 'Eyeliner'])

My goal is to get the key values of the dictionary using the Series. So the result will be

'Facial Care'
'Facial Care'
'Eye Care'
'Eye Care'

Thank you

Upvotes: 1

Views: 22

Answers (1)

Andrej Kesely
Andrej Kesely

Reputation: 195438

Use Series.map with inversed product_type_dic dictionary:

product_type_dic = {
    "Facial Care": ["Facial_cleanser", "Beard_oil"],
    "Eye Care": ["Around-eye_cream", "Eyeliner"],
}


inv_product_type_dic = {vv: k for k, v in product_type_dic.items() for vv in v}
s = pd.Series(["Facial_cleanser", "Beard_oil", "Around-eye_cream", "Eyeliner"])

print(s.map(inv_product_type_dic))

Prints:

0    Facial Care
1    Facial Care
2       Eye Care
3       Eye Care
dtype: object

Upvotes: 1

Related Questions