/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 vote208 views1 answer
Python cv2 read image grayscale
Python
0 vote217 views1 answer
Python pipe select where
Python
0 vote207 views1 answer
Selection sort python
Python
12 votes190 views4 answers
Rotate labels matplotlib
Python
5 votes196 views1 answer
How to use a string variable as a variable name in python
Python
0 vote230 views1 answer
Python beautifulsoup load cookies download file from url
Python
0 vote210 views1 answer
Get context data django
Python
0 vote197 views2 answers
What is fn.call
Python
0 vote195 views1 answer
Pss signatures python
Python
0 vote189 views1 answer
Python How To Convert a String to Variable Name
Python