python - 应用 "group by"后选择第一个或最后一个 n 组

Question

我想在按列应用组后提取特定数量的组。例如前 2 或 3 组。

我有一个数据框:

id gender value1  f      1123 1  f      102  m      1232  m      154  2  m      1653  m      654  3  m      9874  f      76544  f      76544  f      7654...  ...  ....

I want something like this

id gender value2  m      1232  m      154 3  m      654  3  m      987... .. ...  

My code is:

dtFrame2 = dtFrame.groupby('id').head(2)
dtFrameMale = dtFrame2.loc[dtFrame2.gender=='male']
temp = maleGroups.filter(lambda x: len(x) == 2)

最后一条语句为我提供了两行的所有组，但之后我想提取前两个、三个或 n 个组。

Answer 1

像这样

In [60]: s = df[df['gender'] == 'm'].groupby('id').size()

In [61]: s.name = 'size'

In [62]: df2 = df.join(s, on='id')

In [63]: df2[df2['size'] == 2]
Out[63]: 
   id gender  value  size
5   3      m    654     2
6   3      m    987     2