/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 vote90 views1 answer
Snapchat api in python
Python
0 vote123 views1 answer
Django python base 64 encode
Python Django
1 vote104 views1 answer
How to get the size of an object in python
Python
0 vote92 views1 answer
Name 'q' is not defined in jupyter notebook
Python
0 vote109 views1 answer
Key item loop list python
Python
0 vote94 views1 answer
How to convert integer to hex signed 2's complement in python
Python
1 vote102 views1 answer
Python get object class
Python
0 vote95 views1 answer
Even numbers from 1 to 100 in python
Python
2 votes124 views1 answer
Django group with permission
Python Django
0 vote137 views1 answer
Problems with flask bootstrap
Python Flask