在每个熊猫数据框行中查找前n个最高价值列的名称

在每个熊猫数据框行中查找前n个最高价值列的名称

您可以

np.argsort

用来查找每一行的
n个 最大项的索引：

import numpy as npimport pandas as pddf = pd.Dataframe({'id': [1, 2, 3, 4, 5], 'p1': [0, 0, 1, 1, 2], 'p2': [9, 2, 3, 5, 3], 'p3': [1, 3, 10, 3, 7], 'p4': [4, 4, 7, 1, 10]})df = df.set_index('id')nlargest = 3order = np.argsort(-df.values, axis=1)[:, :nlargest]result = pd.Dataframe(df.columns[order], columns=['top{}'.format(i) for i in range(1, nlargest+1)],index=df.index)print(result)

产量

   top1 top2 top3id    1    p2   p4   p32    p4   p3   p23    p3   p4   p24    p2   p3   p15    p4   p3   p2