/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

9 votes109 views6 answers
Dropna pandas
Python
0 vote108 views1 answer
Pytest suppress errors
Python
0 vote143 views1 answer
Python make 1d array from n-d array
Python
1 vote101 views1 answer
Get local python api image url
Python
0 vote117 views1 answer
Numpy remove columns containing nan
Python
1 vote125 views1 answer
Why men are better than woman
Python
0 vote118 views1 answer
How to search for a data in excel pandas
Python
12 votes109 views2 answers
Is python an object oriented language
Python
1 vote110 views2 answers
Python int16
Python
0 vote138 views1 answer
Python flask models user
Python Flask