Найдите имена n крупнейших значений в каждой строке фрейма данных. ⇐ Python

[Anonymous]

Я могу найти n наибольших значений в каждой строке массива numpy (ссылка), но при этом теряется информация о столбце, а это то, что мне нужно. Допустим, у меня есть данные:

Код: Выделить всё

import pandas as pd
import numpy as np

np.random.seed(42)
data = np.random.rand(5,5)
data = pd.DataFrame(data, columns = list('abcde'))
data

a         b         c         d         e
0  0.374540  0.950714  0.731994  0.598658  0.156019
1  0.155995  0.058084  0.866176  0.601115  0.708073
2  0.020584  0.969910  0.832443  0.212339  0.181825
3  0.183405  0.304242  0.524756  0.431945  0.291229
4  0.611853  0.139494  0.292145  0.366362  0.456070

Мне нужны имена крупнейших участников в каждой строке. Таким образом, для n = 2 результат будет следующим:

Код: Выделить всё

0  b  c
1  c  e
2  b  c
3  c  d
4  a  e

Я могу сделать это, пройдя по кадру данных, но это будет неэффективно. Есть ли более питонический способ?

Подробнее здесь: https://stackoverflow.com/questions/754 ... -dataframe