/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

-1 vote274 views2 answers
Matplotlib set colorbar range
Python
0 vote255 views3 answers
Pivot table pandas
Python
0 vote271 views1 answer
Find daily aggregation in pandas
Python
1 vote261 views1 answer
Python yaml load_all
Python
0 vote211 views2 answers
Python fraction simplify
Python
0 vote256 views1 answer
Python enumerate conditional statements
Python
0 vote282 views1 answer
Python dlib
Python
-1 vote241 views1 answer
Pip path windows 10
Python
0 vote243 views10 answers
Python codes and answers cheat code pdf
Python
7 votes286 views3 answers
How to exit venv in python
Python