在此上下文中,您有一个包含分组数据的 pandas DataFrame,并希望从每个组中提取第一行。这可以通过多种方法来实现。
一种简单的方法是利用first()函数,它从每列中检索第一个非空值:
df.groupby('id').first()
此方法排除输出中的索引列。要将 id 列包含为列,请使用 reset_index():
df.groupby('id').first().reset_index()
或者,您可以使用 head(n) 获取每组的前 n 行:
df.groupby('id').head(2).reset_index(drop=True)
在此示例中, drop=True 参数从输出中删除重置索引列。
另一种方法涉及在 groupby() 中设置 as_index 参数:
df.groupby('id', as_index=False).nth(0)
此方法包括 id默认为一列。
请记住根据您对列包含的具体要求和要检索的行数来调整这些方法。
以上是如何提取 Pandas DataFrame 中每组的第一行?的详细内容。更多信息请关注PHP中文网其他相关文章!