鍍金池/ 問答/數(shù)據(jù)分析&挖掘  Python/ 求教一個(gè)dataframe 分組問題

求教一個(gè)dataframe 分組問題

clipboard.png
這樣的一個(gè)表，想用key1分組，取data2大的那個(gè)行，做一個(gè)表怎么做？
試過df.groupby('key1',as_index=False).max()取的是所有列的最大，這樣不行，請(qǐng)教怎么做？
這是原數(shù)據(jù)：
df = pd.DataFrame({'key1':['a','a','b','b','a'],'key2':['one','two','one','two','one'],'data1':np.random.randn(5),'data2':np.random.randn(5)})
df

回答
編輯回答
神經(jīng)質(zhì)
def f(df, col=1):
    return df[df['data2'] == max(df['data2'])]

df1 = df.groupby(['key1']).apply(f)
2017年10月13日 06:37