在我的多索引 pandas dataframe 中，我怎么能 select 只为每个级别（0）的前两行？

Question

say i have a df, and i group by two columns. i then want to only take the first two rows for my grouped by object. ie

grouped_data = df.groupby(['company','person']).first() 

how do i then select the first two rows for each of these. eg for company = asda there are 8 rows i,e, 9 people under this company but i only want the first two rows. how can i do this using the dataframe above? note i have used first because after grouping by i want to retain the column by column information without aggregation.

Answer 1

If you want frist two rows for each companies you can do:

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