在 dataframe 中查找每列的最高 n 值 [重复]

Question

这个问题在这里已经有了答案：

使用 pandas （3 个答案）按列返回 n 个最小索引

去年关闭。

我想在 dataframe 中找到每列的最高 3 个值，并返回按值排序的索引名称。 dataframe 看起来像这样：

df = pd.DataFrame({"u1":[1,2,-3,4,5],
                   "u2":[8,-4,5,6,7],
                   "u3":[np.NaN,np.NaN,np.NaN,np.NaN,np.NaN]},
                   index=["q1","q2","q3","q4","q5"])

结果将如下所示：

u1   u2   u3
q5   q1   NaN
q4   q5   NaN
q2   q4   NaN

Answer 1

您可以将apply与pandas.Series.nlargest function 一起使用。

df.apply(lambda x: pd.Series(x.nlargest(3).index))
   u1  u2   u3
0  q5  q1  NaN
1  q4  q5  NaN
2  q2  q4  NaN

Answer 2

我们需要先做stack然后sort_values并使用cumcount和pivot创建新索引来重塑它

s=df.stack().reset_index().sort_values(0,ascending=False).\
          assign(index=lambda x : x.groupby('level_1').cumcount()).\
           pivot('index','level_1','level_0').\
            reindex(columns=df.columns).head(3)
Out[308]: 
       u1  u2  u3
index            
0      q5  q1 NaN
1      q4  q5 NaN
2      q2  q4 NaN