分组并选择每组的第一个、第二个和第四个成员

Question

相关：pandas数据帧分组并获取第n行

我可以使用groupby方法，并使用选择前N个组成员

df.groupby('columnA').head(N) 

但是，如果我想要每组的第一、第二和第四名成员呢？

Answer 1

GroupBy.nth需要一个列表，所以您可以只执行

df = pd.DataFrame({'A': list('aaaabbbb'), 'B': list('abcdefgh')})
df.groupby('A').nth([0, 1, 3])
B
A   
a  a
a  b
a  d
b  e
b  f
b  h

# To get the grouper as a column, use as_index=False
df.groupby('A', as_index=False).nth([0, 1, 3])
A  B
0  a  a
1  a  b
3  a  d
4  b  e
5  b  f
7  b  h

Answer 2

您可以进行

df.groupby('columnA').apply(lambda x : x.iloc[[has to 0,1,3],:]).reset_index(level=0,drop=True)

Answer 3

df1 = df.groupby('columnA').head(4) 
df1.drop(df.groupby('columnA').head(4).index.values[2], axis=0)