獲取列表中最大N個數的索引

• 前言
• 一、使用步驟
• • 1.引入庫
  • 2.編寫函數

---

前言

我們在處理數據過程中，經常需要查詢list中最大的N個數對應的索引，目前的一些方法應對具有重復值的list會發生諸多問題，比如索引重復等等，那么我們如何快速得到list中最大N個數的索引值呢？

---

基于pandas的索引查詢方法

一、使用步驟

1.引入庫

代碼如下（示例）：

import pandas as pd

2.編寫函數

代碼如下（示例）：

def Get_List_Max_Index(list_, n):"""function：計算列表中最大的N個數對應的索引Parameters:list_ - 要分析的列表(list)n - 截取最大的n個數(int)Returns:n_index - 最大n個數的索引Modify:2020-11-23"""N_large = pd.DataFrame({'score': list_}).sort_values(by='score', ascending=[False])return list(N_large.index)[:n]

我們可以按如下代碼使用該程序

>>> result = Get_List_Max_Index([1, 2, 3, 4, 4, 4, 5, 5, 6], 5) >>> result [8, 6, 7, 3, 4]

---

總結

以上是生活随笔為你收集整理的获取列表中最大N个数的索引的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。