假设我有一个df，并且我按两列分组。然后，我只想取我的groupedby对象的前两行。即
grouped_data = df.groupby(['company','person']).first()
然后我该如何为每一行选择前两行。例如，对于company=asda，这家公司有8行i，e，9个人，但我只想要前两行。如何使用上面的数据帧来完成此操作？注意，我首先使用了，因为在分组之后，我希望保留逐列的信息，而不进行聚合。
如果你想为每个公司设置前两行，你可以这样做：
df.groupby('company').head(2)