熊猫分组依据:无法保留原始数据框分组依据

时间:2019-12-03 10:12:42

标签: python pandas

这个问题可能很简单。我有一个数据框df,如下所示

DealerID    ZMONTH    ZREGION    Cust_Cat     ZBAS_QTY   ZBAS_VAL
1001        201905    ABC        M            200        750
1001        201906    ABC        N            300        480
1001        201907    NOP        P            800        1156
1002        201905    PQR        M            350        525
1002        201906    PST        M            480        690
1002        201907    SNP        P            200        780

我想应用df.groupby(),以便为每个DealerID获得具有max() ZBAS_VAL的行。换句话说,结果数据框应该看起来像

DealerID    ZMONTH    ZREGION    Cust_Cat     ZBAS_QTY   ZBAS_VAL
1001        201907    NOP        P            800        1156
1002        201907    SNP        P            200        780

到目前为止,我的方法:

df = df.groupby(['DealerID'])['ZBAS_VAL'].max().reset_index()

但是,这种方法仅返回ZBAS_VALDealerID列。我想要的是所有其他剩余的列。

有任何线索吗?

0 个答案:

没有答案