说我有一个df，我按两列分组。然后，我只想按对象对我的前两行进行分组。即
grouped_data = df.groupby(['company','person']).first()
然后我该如何为每一个选择前两行。例如对于company = asda，有8行，即该公司下有9个人，但我只希望前两行。我如何使用上面的数据框来做到这一点？请注意，我之所以先使用过，是因为在按分组后，我希望按列信息保留而不进行汇总。
答案 0 :(得分：1)
如果您希望每个公司第一排两行，您可以执行以下操作：
df.groupby('company').head(2)