如何有效率地取得 Pandas DataFrame 中每組的前幾筆記錄？

Pandas：高效獲取組內最靠前的記錄

使用Pandas DataFrame 時，經常需要從每個組中提取前導記錄。常見的方法是利用 'groupby' 和 'apply' 函數來列舉每個群組內的記錄。

dfN = df.groupby('id').apply(lambda x:x['value'].reset_index()).reset_index()

但是，有一個更簡化的方法：

df.groupby('id').head(2)

這方法直接取得最上面的記錄，無需中間計算。此外，產生的 DataFrame 保留其原始索引。

要展平產生的MultiIndex，請使用：

df.groupby('id').head(2).reset_index(drop=True)

這將產生以下DataFrame：

id	value
1	1
1	2
2	1
2	2
3	1
4	1

或者，您可以使用SQL 的「row_number( )」視窗函數來有效率地列舉組內的記錄。然而，此功能目前在 Pandas 中不可用。

以上是如何有效率地取得 Pandas DataFrame 中每組的前幾筆記錄？的詳細內容。更多資訊請關注PHP中文網其他相關文章！