在线观看不卡亚洲电影_亚洲妓女99综合网_91青青青亚洲娱乐在线观看_日韩无码高清综合久久

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

求教一個dataframe 分組問題

clipboard.png
這樣的一個表,想用key1分組,取data2大的那個行,做一個表怎么做?
試過df.groupby('key1',as_index=False).max()取的是所有列的最大,這樣不行,請教怎么做?
這是原數(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