/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 vote210 views1 answer
If list is null python apply any function site:stackoverflow.com
Python
0 vote208 views1 answer
Abstract user in django
Python Django
1 vote167 views1 answer
Python change name of the imported function
Python
37 votes160 views6 answers
If list element contains string python
Python
0 vote191 views1 answer
Maximum number of charectors allowed for a string variable in python
Python
1 vote153 views1 answer
Filter dataframe columns vy a list of columns
Python
0 vote184 views1 answer
Travis deployment script for django applications to heroku
Python
13 votes196 views2 answers
Min int python
Python
0 vote164 views1 answer
Why am i not able to import wtf flask
Python
0 vote169 views1 answer
Markers are not visible on line plot
Python