/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 vote142 views1 answer
How to plot multiple images in python
Python
10 votes193 views5 answers
Check gpu in tensorflow
Python
3 votes206 views2 answers
AttributeError: module 'tensorflow' has no attribute 'placeholder'
Python
0 vote183 views1 answer
TypeError: add() got an unexpected keyword argument 'kernel_regularizer'
Python
0 vote198 views1 answer
Django today date in template
Python
0 vote148 views1 answer
Tqdm pandas apply
Python
19 votes211 views6 answers
Pip install django
Python
0 vote195 views1 answer
How to add list as a column to dataframe
Python
0 vote183 views1 answer
How do i select a range of columns by index
Python
0 vote239 views1 answer
Python collections cheat sheet
Python