如何在一列中找到对应的值，由于其他列从另一列组的最大值?

Question

由于数据帧:

tbody> <<tr>

COUNTRY_CODE	PDP_SOURCE	TREAT_SLIT	ATC_METRIC
AR	通道	ex_1	12.78
AR	通道	ex_2	11.28
AR	通道	ex_3	11.96
AR	最爱	ex_1	12.78
AR	最爱	ex_2	12.28
AR	最爱	ex_3	13.96
最爱	ex_1	11.2
最爱	ex_2	10.28
最爱	ex_3	10.96

这是一种方法

# use groupby to get the index where ATC_METRIC is max, then 
# use loc to return those rows
df.loc[df.groupby(['COUNTRY_CODE','PDP_SOURCE'] )['ATC_METRIC'] .idxmax()]

结果

COUNTRY_CODE    PDP_SOURCE  TREAT_SLIT  ATC_METRIC
0             AR    Aisles            ex_1  12.78
5             AR    Favorites         ex_3  13.96
6             BR    Favorites         ex_1  11.20