獲取數據框中列中 n 個最大值的索引

Question

我有一個數據框，我想獲取每行中 4 個最大值的索引和值。 例如，在下面的df中，a列中，10、6、7、8是四個最大值。

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

我想要的輸出是：

Answer 1

嘗試最大，

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

輸出：

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

Answer 2

您可以對a列進行排序

out = (df.sort_values('a', ascending=False).iloc[:4]
       .sort_index(ascending=True)
       .reset_index())

print(out)

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