/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

4 votes186 views2 answers
Print format python
Python
10 votes196 views4 answers
Pandas
Python
17 votes179 views2 answers
Random chars generator python
Python
0 vote188 views1 answer
Python array find lambda
Python
0 vote156 views2 answers
Nested renamer is not supported pandas
Python
11 votes204 views4 answers
Add conda env to jupyter
Python
0 vote199 views1 answer
How to update values in tkinter
Python
0 vote177 views1 answer
Python strip txt
Python
1 vote205 views1 answer
Python transpose np array
Python
2 votes234 views2 answers
AttributeError: module 'django.contrib.auth.views' has no attribute 'login'
Python Django