溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本篇內容介紹了“Pandas中怎么分組再取N項”的有關知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領大家學習一下如何處理這些情況吧！希望大家仔細閱讀，能夠學有所成！
在 pandas 中，DataFrame 是我們經常用到的工具。有時候，我們可能會需要對數據按某個字段進行分組，然后每個組取N項。例如：
現在，我想每個職位任取三個用戶。
相信有同學會使用 for 循環，依次循環每一行，每個職位選3個，存入一個臨時的列表里面。循環完成以后再轉成一個新的 DataFrame。但這個方式顯然不夠智能。
那么，我們有沒有什么辦法能夠不使用循環就做到這一步呢?也許有同學想到了使用 groupby。我們來看看效果。
看起來僅僅是統計了每個職位的數量。那么，如何才能保留所有字段呢?
實際上我們可以把.size()改成.head(3):
看起來這里的.head(3)似乎沒有什么作用。這個時候，我們思考一下 Python 里面，如果要使用itertools.groupby，官方文檔里面有這樣一段話：
Generally, the iterable needs to already be sorted on the same key function.
如下圖所示：
這段話告訴我們，要使用itertools.groupby，我們需要提前對被分組的字段進行排序。
那么，我們試一試在如果提前對 DataFrame 進行排序，然后再 groupby 會怎么樣：
成功了。每個職位都取了3個。
可能大家發現最左邊的索引是亂序，看起來不好看。那么我們還可以重設一下索引：
“Pandas中怎么分組再取N項”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站，小編將為大家輸出更多高質量的實用文章！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
