/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

50 votes147 views3 answers
Python read arguments
Python
25 votes126 views6 answers
Python if
Python
0 vote138 views1 answer
Icon tkiner
Python
2 votes129 views1 answer
Why is python hard
Python
1 vote135 views1 answer
Exclude first value of an array python
Python
51 votes123 views4 answers
Get date from datetime python
Python
0 vote131 views1 answer
Prime number checking algorithm
Python
2 votes136 views1 answer
How to update pandas
Python
14 votes132 views3 answers
Create channel
Python
0 vote124 views1 answer
Tuple and dictionary in python
Python