/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 vote265 views1 answer
How to go up levels in path python
Python
0 vote227 views1 answer
Python dataframe copy structure
Python
0 vote211 views1 answer
Removing new line character in python from dataframe
Python
5 votes221 views2 answers
Encrypt and decrypt python
Python
1 vote191 views1 answer
User defined functions python
Python
2 votes203 views1 answer
Load python file in jupyter notebook
Python
3 votes150 views1 answer
Pandas date missing from datetime
Python
0 vote562 views1 answer
Fibonacci sequence python genorator
Python
22 votes197 views4 answers
How to create a string in python
Python
11 votes220 views2 answers
Change pandas column value based on condition
Python