首頁 >後端開發 >Python教學 >如何有效率地取得每個Pandas Group內的前N筆記錄?

如何有效率地取得每個Pandas Group內的前N筆記錄?

Patricia Arquette
Patricia Arquette原創
2024-12-02 19:27:14973瀏覽

How to Efficiently Get the Top N Records within Each Pandas Group?

取得Pandas 組內的最高記錄

在以下資料集中:

我們希望獲得每個id 的前兩筆記錄。一個簡單的方法是使用groupby 方法在每個群組內分配行號:

但是,head 函數提供了更有效的解決方案:

此操作會產生:

要刪除MultiIndex並展平結果,使用:

這會產生所需的輸出:

因此,head 函數提供了一種簡潔且最佳化的方法來檢索每個Pandas 組中最上面的記錄。

以上是如何有效率地取得每個Pandas Group內的前N筆記錄?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn