填寫Nan基於組

Question

我想根據列值的平均值填寫NaN。 例如：

 Groups  Temp 

1  5   27 
2  5   23 
3  5   NaN (will be replaced by 25) 
4  1   NaN (will be replaced by the mean of the Temps that are in group 1) 

有什麼建議嗎？謝謝 ！

Answer 1

使用groupby，transfrom，並與fillna和mean lamdba功能：

df = df.assign(Temp=df.groupby('Groups')['Temp'].transform(lambda x: x.fillna(x.mean()))) 
print(df) 

輸出：

Temp 
0 27.0 
1 23.0 
2 25.0 
3 NaN