/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

0 vote113 views1 answer
How to create a palindrome prime in python 3
Python
3 votes127 views2 answers
Python dict order a dict by key
Python
0 vote119 views1 answer
How to use rbind() to combine dataframes
Python
12 votes113 views10 answers
Timer
Python
0 vote167 views1 answer
Counter in sort python
Python
0 vote252 views1 answer
Pandas index from 1
Python
0 vote135 views1 answer
Taggablemanager serializer django
Python Django
1 vote150 views1 answer
Django migrate fake zero
Python Django
1 vote137 views1 answer
Pip uninstalled itself
Python
1 vote118 views6 answers
Insertion sort
Python