如何使用nlargest分组并保留所有列?

时间:2019-05-11 14:14:30

标签: python python-3.x pandas-groupby

我想对DataFrame进行分组,并获取列“ C”的最大数据。 而返回的是系列,而不是DataFrame。

dftest = pd.DataFrame({'A':[1,2,3,4,5,6,7,8,9,10],
                       'B':['A','B','A','B','A','B','A','B','B','B'],
                       'C':[0,0,1,1,2,2,3,3,4,4]})
dfn=dftest.groupby('B',group_keys=False)\
            .apply(lambda grp:grp['C'].nlargest(int(grp['C'].count()*0.8))).sort_index()

结果得到一个序列。

2    1
4    2
5    2
6    3
7    3
8    4
9    4
Name: C, dtype: int64

我希望结果是DataFrame,就像

    A  B  C
2   3  A  1
4   5  A  2
5   6  B  2
6   7  A  3
7   8  B  3
8   9  B  4
9  10  B  4  

******更新************** 抱歉,'A'列实际上不是序列整数,dftest可能更像

dftest = pd.DataFrame({'A':['Feb','Flow','Air','Flow','Feb','Beta','Cat','Feb','Beta','Air'],
                       'B':['A','B','A','B','A','B','A','B','B','B'],
                       'C':[0,0,1,1,2,2,3,3,4,4]})

结果应该是

    A     B  C
2   Air   A  1
4   Feb   A  2
5   Beta  B  2
6   Cat   A  3
7   Feb   B  3
8   Beta  B  4
9   Air   B  4 

2 个答案:

答案 0 :(得分:0)

这可能有点笨拙,但它确实满足您的要求:

dfn= dftest.groupby('B').apply(lambda 
grp:grp['C'].nlargest(int(grp['C'].count()*0.8))).reset_index().rename(columns= 
{'level_1':'A'})
dfn.A = dfn.A+1
dfn=dfn[['A','B','C']].sort_values(by='A')

答案 1 :(得分:0)

感谢我的朋友,以下代码对我有用。

dfn=dftest.groupby('B',group_keys=False)\
            .apply(lambda grp:grp.nlargest(n=int(grp['C'].count()*0.8),columns='C').sort_index())

dfn是

In [8]:dfn
Out[8]: 
    A  B  C
2   3  A  1
4   5  A  2
6   7  A  3
5   6  B  2
7   8  B  3
8   9  B  4
9  10  B  4

我以前的代码处理系列,后一个代码处理DataFrame。