python - Pandas Group By：无法保留原始数据框发布 groupby

Question

这个问题可能是微不足道的。我有一个数据框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_VAL和DealerID列。我想要的是所有其他剩余的列。

有什么线索吗？

标签： pythonpandas