elegant way to replace multiple list of values with a multiple single value

Question

I have a dataframe like as shown below

df = pd.DataFrame()
df['text'] = ['p', 'S', 'primary','PRI','SECONDARY', 'SEC', 'S', 'TERTIARY','T','third']

I would like to replace a list of values like as shown below

a) Replace P, PRIMARY,PRI with primary b) Replace S, SECONDARY, SEC with secondary c) Replace T, TERTIARY, THIRD with third

I tried the below

df['text'] = df['text'].replace(['P','PRIMARY','PRI'],'primary')
df['text'] = df['text'].replace(['S','SECONDARY','SEC'],'secondary')
df['text'] = df['text'].replace(['T','TERTIARY','THIRD'],'tertiary')

But is there any other efficient and elegant way to write this in a single line?

I expect my output to be like as shown below

     text
0   primary
1   secondary
2   primary
3   primary
4   secondary
5   secondary
6   secondary
7   tertiary
8   tertiary
9   tertiary

jezrael · Accepted Answer · 2021-04-13 07:29:31Z

2

One idea for avoid multiple replace is use dictionary and flatten it to another dict for keys from lists values, for match convert column to uppercase by Series.str.upper:

d = {'primary': ['P','PRIMARY','PRI'],
     'secondary':['S','SECONDARY','SEC'],
     'tertiary':['T','TERTIARY','THIRD']}


d1 = {x: k for k, v in d.items() for x in v}
df['text'] = df['text'].str.upper().replace(d1)
print (df)
        text
0    primary
1  secondary
2    primary
3    primary
4  secondary
5  secondary
6  secondary
7   tertiary
8   tertiary
9   tertiary

edited Apr 13, 2021 at 7:29

answered Apr 13, 2021 at 7:18

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

The Great Over a year ago

is that case-insensitive?

jezrael Over a year ago

@TheGreat - Unfortuantely not, need convert it to uppercase.

Collectives™ on Stack Overflow

elegant way to replace multiple list of values with a multiple single value

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related