/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

7 votes151 views4 answers
How to create 2d list in python
Python
4 votes139 views2 answers
Django how to delete a db field
Python
0 vote193 views1 answer
Settings.debug django
Python Django
8 votes173 views3 answers
Import turtle as t
Python
1 vote170 views1 answer
Python index 2d array
Python
17 votes155 views4 answers
How to get int input in python
Python
0 vote141 views1 answer
Detailview
Python
3 votes151 views1 answer
Streamlit ssl error
Python
0 vote165 views1 answer
Python remove header
Python
0 vote154 views2 answers
Set operators
Python