pandas数据框：在按另一列分组后获取最大列数

Question

我尝试为每个A获得最大值B. C和D在那里，因为我的数据集不仅仅是我要排序的2列，而是获得最大值。

import pandas
import numpy

df = pandas.DataFrame({'A': [10, 10, 20, 20, 30, 20, 10, 20],
            'B': [1001, 1002, 2002, 2003, 3001, 2003, 1002, 2003],
            'C': numpy.random.randn(8),
            'D': numpy.random.randn(8)})

这样的事情：

df[['A', 'B']].somepandas(magic)

期望的结果是：

   B
A        
10 1002
20 2003
30 3001

但到目前为止，我只知道在没有分组的情况下获得每列的最大值：

df[['A', 'B']].max(axis=0)

A      30
B    3001
dtype: int64

感谢任何想法

Answer 1

您需要groupby'A'列，然后选择'B'列并在列上调用max()：

In [42]:

df.groupby('A')['B'].max()
Out[42]:
A
10    1002
20    2003
30    3001
Name: B, dtype: int64

您可以一次在不同的列上执行多项功能，请参阅文档：http://pandas.pydata.org/pandas-docs/stable/groupby.html#aggregation