python - pandas:如何选择每个 GROUP BY 组中的第一行？

Question

与Select first row in each GROUP BY group?基本相同仅在 Pandas 中。

df = pd.DataFrame({'A' : ['foo', 'foo', 'foo', 'foo', 'bar', 'bar', 'bar', 'bar'],
                'B' : ['3', '1', '2', '4','2', '4', '1', '3'],
                    })

排序看起来很有希望:

df.sort('B')

     A  B
1  foo  1
6  bar  1
2  foo  2
4  bar  2
0  foo  3
7  bar  3
3  foo  4
5  bar  4

但是 first 不会给出想要的结果...... df.groupby('A').first()

     B
A     
bar  2
foo  3

Answer 1

通常，如果您希望您的数据按分组方式排序，但它不是要分组的列之一，那么最好使用 sort执行 groupby 之前的 df:

In [5]:
df.sort_values('B').groupby('A').first()

Out[5]:
     B
A     
bar  1
foo  1