Pandas groupby 将定制的 function 应用到每个组

Question

I have a table with multiple columns. For one of the columns AC , I need to replace it with Result AC as shown in the table below

The function has a pseudo code as below:

For each unique address:
If 
# of unique(AC) < Value in column B AND 
len(Top 1 (unique(AC))) != len(Top 2 (unique(AC))):
return mode(air_conditioning) 

Else
For each unique(AC), calculate sum(area)
If Top 1 (sum(area)) > Top 2 (SUM(area)):
return air_conditioning which has Top 1 (SUM(building_area))

Else
For each unique(AC), SUM(value)
return air_conditioning with Top 1 (SUM(value))
 

Answer 1

Groupby, transfrom sum should do

df['col2']=df.groupby('col1')['col2'].transform('sum')