[FIXED] Get the index of n maximum values in a column in dataframe

Issue

I have a data frame and I want to get the index and value of the 4 maximum values in each rows. For example, in the following df, in column a, 10, 6, 7, 8 are four maximum values.

import pandas as pd
df = pd.DataFrame()
df['a'] = [10, 2, 3, -1,4,5,6,7,8]
df['id'] = [100, 2, 3, -1,4,5,0,1,2]
df

The output which I want is:

Thanks for your help.

Solution

Try nlargest,

df.nlargest(4, 'a').reset_index()

Output:

   index   a   id
0      0  10  100
1      8   8    2
2      7   7    1
3      6   6    0

Answered By - Scott Boston