/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 vote355 views1 answer
Attributeerror: module
Python
6 votes336 views3 answers
Concatenate lists
Python
1 vote332 views1 answer
Get first element of tuple python
Python
8 votes325 views2 answers
How to check if a word is in a dictionary
Python
3 votes310 views2 answers
Module 'cv2' has no 'imread' member
Python
0 vote296 views1 answer
Make exact copy of an environment
Python
5 votes350 views1 answer
Get index in foreach py
Python
0 vote280 views1 answer
Restart computer py
Python
1 vote291 views1 answer
Decrease hours in datetime python
Python
0 vote323 views2 answers
Jupyter notebook print string with variables
Python