python - 如何获取单行数据框中与 n 个最大值对应的列？

Question

我有一个这样的数据框

    c0   c1   c2   c3   c4   c5
0   2    4    3    1    6    5

并希望获取与单行中的 n 最大值对应的列，例如n=3

    c4   c5   c1
0   6    5    4

我怎样才能最有效地实现这一点(w.r.t. 内存使用、执行时间和与 dask 并行化的能力)？

作为引用 w.r.t.性能 @stark 的答案(500 列的数据框，n=10):使用 df.sort_values(0, axis='columns', ascending=False).iloc[:, :n] I使用 %%timeit 获得 649 µs。

Answer 1

使用 numpy - np.argsort对于排序值的位置，获取前 3 个值 使用和 iloc 选择:

df = df.iloc[:, np.argsort(-df.values[0])[:3]]
print (df)
   c4  c5  c1
0   6   5   4