/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 votes32 views4 answers
If list item in string python
Python
0 vote28 views1 answer
How to define the range of values in seaborn heatmap
Python
1 vote30 views1 answer
How to delete print statement from console pythonn
Python
0 vote32 views2 answers
PHP echo multiple lines example
Python
0 vote30 views1 answer
How to set time limit for receiving data in socket python
Python
1 vote29 views1 answer
Django login redirect
Python
0 vote37 views1 answer
Get last save id django model
Python Django
-1 vote28 views1 answer
Python squared
Python
3 votes49 views2 answers
Wap in python to check a number is palindrome or not
Python
0 vote28 views1 answer
Strip array of strings python
Python