python - 如何根据 Pandas 中的另一列聚合一列

Question

    year   fruit  sales
0  2010   Apple     10
1  2011   Apple     20
2  2010  Banans     50000
3  2011  Banans     30

我想要的是这样的:

    fruit   min_year  sales_2010  max_year  sales_2011
0   Apple       2010          10      2011          20
1   Banans      2010       50000      2011          30

Answer 1

首先使用DataFrame.add_suffix将min和max聚合到df1 ，然后按 DataFrame.pivot 进行旋转与 DataFrame.add_prefix最后通过 concat 一起加入:

df1 = df.groupby('fruit')['year'].agg(['min','max']).add_suffix('_year')
df2 = df.pivot('fruit','year','sales').add_prefix('sales_')

df = pd.concat([df1, df2], axis=1)
print (df)
        min_year  max_year  sales_2010  sales_2011
fruit                                             
Apple       2010      2011          10          20
Banans      2010      2011       50000          30