/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

13 votes64 views1 answer
5.4.7 categories python
Python
3 votes123 views2 answers
Normal distribution
Python
4 votes108 views1 answer
Python random true false
Python
1 vote114 views1 answer
How to get files list from active directory from where the python script is running
Python
-1 vote101 views1 answer
How to give values to all users with discord api python grepper
Python
39 votes109 views6 answers
List to string
Python
1 vote129 views1 answer
Check if can convert to float python
Python
0 vote123 views1 answer
Python glob subdirectories
Python
1 vote189 views2 answers
Selenium text value is empty in flask returns
Python
0 vote99 views1 answer
UTC to ISO 8601:
Python