/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

1 vote313 views1 answer
Python logging to console exqmple
Python
-2 votes320 views2 answers
Django form date picker
Python Django
0 vote336 views1 answer
Set title name in steamlit 0.790
Python
0 vote304 views6 answers
Python list Comprehensions
Python
0 vote299 views2 answers
Install socketio flask
Python
11 votes291 views3 answers
Change column order dataframe python
Python
0 vote282 views1 answer
Remove base from terminal anaconda
Python
15 votes302 views1 answer
Api in python
Python
114 votes350 views5 answers
How to use inputs in python
Python
0 vote325 views1 answer
Non preemptive priority scheduling in c#
Python