/Python/Def conditional_impute(input_df, choice='median')

Def conditional_impute(input_df, choice='median')

def conditional_impute(input_df, choice='median'):
    new_df = input_df.copy()
    
    if choice == 'median':
        new_df['Age'] = round(new_df.groupby(['Sex', 'Pclass'])['Age'].transform(func = lambda x: x.fillna(x.median())),1)
        
    elif choice == 'mean':
        new_df['Age'] = round(new_df.groupby(['Sex', 'Pclass'])['Age'].transform(func = lambda x: x.fillna(x.mean())),1)
        
    else:
        raise ValueError('Please choose either median or mean as your impute choice.')
    
    return new_df

Source: stackoverflow.com

Def conditional_impute(input_df, choice='median')

if choice == 'median':
        new_df['Age'] = new_df.groupby(['Sex', 'Pclass'])['Age'].transform(func = lambda x: x.fillna(round(x.median(),1)))

Source: stackoverflow.com

Similar codes

2 votes131 views1 answer
Pandas find top 10 values in column
Python
1 vote136 views1 answer
Pandas timedelta to seconds
Python
3 votes86 views1 answer
Torch mse loss
Python
0 vote128 views1 answer
Df groupby loop
Python
-1 vote131 views1 answer
Matrix python math
Python
0 vote120 views1 answer
Python multidimensional dictionary
Python
0 vote120 views1 answer
Change column row box colour tkinter
Python
36 votes127 views4 answers
Print colored text to console python
Python
0 vote79 views1 answer
Create bbox R sp
Python
22 votes133 views2 answers
Reset_index(drop=true)
Python