/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

3 votes298 views1 answer
How to get 2 random inputs in a list using for loop
Python
2 votes280 views1 answer
Normalise list python
Python
0 vote279 views1 answer
Concatenação de Strings
Python
1 vote274 views1 answer
Django print settings
Python
3 votes274 views1 answer
Accessing items of tuple in python
Python
0 vote259 views1 answer
Ai chatbot python
Python
16 votes286 views2 answers
Auto clicker in python
Python
-1 vote320 views1 answer
Uninstall python3.8 ubuntu
Python
0 vote285 views1 answer
How to do welcome in bubble letters in python
Python
1 vote273 views1 answer
Code for python shell 3.8.5 games
Python