执行groupby操作时如何保留未使用的列

Question

我有以下数据框，

    restaurant                  ProductKey  QuantityOrdered
0   1760 POLK STREET RESTAURANT 000055-1    425.0
1   1760 POLK STREET RESTAURANT 000246-1    43.0
2   1760 POLK STREET RESTAURANT 000249-1    4.0
3   1760 POLK STREET RESTAURANT 000310-1    30.0
4   1760 POLK STREET RESTAURANT 000606-1    200.0

我需要找到每个餐厅的最大订购量。

我尝试如下使用groupby，

df = df.groupby(['Name'], as_index=True).agg({'QuantityOrdered': 'max', })

我的输出是这样的，

                            QuantityOrdered
restaurant  
1760 POLK STREET RESTAURANT         450.00
4505 ***OAKLAND***                  1441.00
4505 BURGERS & BBQ- SAN FRANCISCO   2865.75
ALX GASTROPUB                       275.00
AMERICANO                           1517.00

有什么方法可以让我获得最大值并保留相关的ProductKey吗？像这样的东西。

                        QuantityOrdered      ProductKey
restaurant  
1760 POLK STREET RESTAURANT         450.00   000055-1 
4505 ***OAKLAND***                  1441.00  000246-1
4505 BURGERS & BBQ- SAN FRANCISCO   2865.75  000606-1  
ALX GASTROPUB                       275.00   000606-1 
AMERICANO                           1517.00  000249-1