Pandas 按兩列分組並獲得最大值

Question

Pandas 按兩列分組並獲得最大值

我用多索引對數據進行了分組

                        
    Model  VehicleType  VehicleType            
    100    sedan                278
           wagon                109
           coupe                  2
           convertible            1
    145    small                 19
    ...                         ...
    zafira sedan                 22
           small                 11
           suv                    7
           convertible            1
           coupe                  1

我需要使用相應的 model 和 VehicleType 來獲取計數的最大值（右列），如下所示：

                        
    Model  VehicleType  VehicleType            
    100    sedan                278
    145    small                 19
    ...                         ...
    zafira sedan                 22
           

感謝您的解決方案！

Answer 1

使用DataFrameGroupBy.idxmax按最大值索引索引，然后按 DataFrame.loc 使用DataFrame.loc ：

df = df.loc[df.groupby(level=0)['VehicleType'].idxmax()]
print (df)
                    VehicleType
Model  VehicleType             
100    sedan                278
145    small                 19
zafira sedan                 22