Re-define dataframe index with map function

Question

I have a dataframe like this. I wanted to know how can I apply map function to its index and rename it into a easier format.

df = pd.DataFrame({'d': [1, 2, 3, 4]}, index=['apple_017', 'orange_054', 'orange_061', 'orange_053'])
df
             d
apple_017    1
orange_054    2
orange_061    3
orange_053    4

There are only two labels in the indeces of the dataframe, so it's either apple or orange in this case and this is how I tried:

data.index = data.index.map(i = "apple" if "apple" in i else "orange")

(Apparently it's not how it works)

Desired output:

         d
apple    1
orange    2
orange    3
orange    4

Appreciate anyone's help and suggestion!

Anurag Dabas · Accepted Answer · 2021-07-14 12:46:27Z

1

Try via split():

df.index=df.index.str.split('_').str[0]

OR

via map():

df.index=df.index.map(lambda x:'apple' if 'apple' in x else 'orange')

output of df:

        d
apple   1
orange  2
orange  3
orange  4

edited Jul 14, 2021 at 12:46

answered Jul 14, 2021 at 12:33

Anurag Dabas

24.3k9 gold badges25 silver badges41 bronze badges

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

1 Comment

RubberDuckProducer Over a year ago

Thanks for helping out! That's exactly what I want

Collectives™ on Stack Overflow

Re-define dataframe index with map function

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related