Pandas custom sorting of the rows based on a list

Question

Given a dataframe and a list, with the values of a column of a dataframe. (the list is equaly long as the rows of the dataframe and each value appear exactly ones) How can i sort the rows in the dataframe according to the order in the list?

import pandas as pd
df = pd.DataFrame({'user': ['Bob', 'Jane', 'Alice'], 
                   'income': [40000, 50000, 42000]})
z = ["Jane", "Alice", "Bob"]

BENY · Accepted Answer · 2020-05-09 14:22:12Z

1

Let us do pd.Categorical with argsort

df=df.iloc[pd.Categorical(df.user,z).argsort()]
df
    user  income
1   Jane   50000
2  Alice   42000
0    Bob   40000

Or reindex

df=df.set_index('user').reindex(z).reset_index()

answered May 9, 2020 at 14:22

BENY

324k22 gold badges176 silver badges250 bronze badges

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

Collectives™ on Stack Overflow

Pandas custom sorting of the rows based on a list

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related