调用nlargest取前N大值时如何返回DataFrame对应行全部列

解决方案 #

直接对整个DataFrame调用nlargest方法即可，不需要提前单独提取maxDD列，代码如下：

fiveLargestDD = outputDF.nlargest(n=5, columns='maxDD').reset_index(drop=True)

说明 #

• DataFrame.nlargest()方法通过columns参数指定按maxDD列筛选前5大值，会直接返回对应行的全部列数据
• 末尾追加.reset_index(drop=True)是为了清除原DataFrame的行索引，满足不需要返回索引的要求
• 如果存在多个相同maxDD值的行，可以额外加keep参数调整筛选逻辑：
  • keep='first'：默认值，保留先出现的行
  • keep='last'：保留后出现的行
  • keep='all'：保留所有符合条件的行，返回结果行数可能超过5

内容的提问来源于stack exchange，提问作者Ivan