熊猫:选择前n组

实例

data = {'a': {0: 'i1', 1: 'i1', 2: 'i2', 3: 'i2', 4: 'i2', 5: 'i3'},
        'b': {0: 't1', 1: 't2', 2: 't3', 3: 't1', 4: 't3', 5: 't2'}}
df = pd.DataFrame(data)

密码

lst = df['a'].value_counts()[:2].index
out = df[df['a'].isin(lst)]

out

     a  b
0   i1  t1
1   i1  t2
2   i2  t3
3   i2  t1
4   i2  t3

如果你想按数量排序。使用以下代码

lst = df['a'].value_counts()[:2].index
m = pd.Series(range(0, 2), index=lst)
out = df[df['a'].isin(lst)].sort_values('a', key=lambda x: m[x])

出来

    a   b
2   i2  t3
3   i2  t1
4   i2  t3
0   i1  t1
1   i1  t2