如何高效地获取 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中文网其他相关文章！