开发者社区 问答 正文

如何得到列中前n个最大值对应的索引?

如何得到列中前n个最大值对应的索引?

展开
收起
游客y244y7ln2rlpa 2021-12-05 20:22:56 321 分享 版权
1 条回答
写回答
取消 提交回答
  • df = pd.DataFrame(np.random.randint(1, 15, 15).reshape(5,-1), columns=list('abc'))
    print(df)
    # 取'a'列前3个最大值对应的行
    n = 5
    df['a'].argsort()[::-1].iloc[:3]
    
    #>	    a   b   c
    	0   5   5   2
    	1  12   7   1
    	2   5   2  12
    	3   5  14  12
    	4   1  13  13
    
    #>	4    1
    	3    3
    	2    2
    	Name: a, dtype: int64
    
    
    2021-12-05 22:22:58
    赞同 展开评论
问答分类:
问答地址: