您可以
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
