/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 vote168 views1 answer
Flask session auto logout in 5 mins
Python
0 vote149 views1 answer
Dataframe, groupby, select one
Python
0 vote140 views1 answer
Python variable is not none
Python
3 votes157 views1 answer
Python get item from queue
Python
0 vote172 views2 answers
Groupby and sort python
Python
4 votes150 views1 answer
Python random true false
Python
1 vote117 views1 answer
Python pattern matching
Python
2 votes152 views1 answer
.env files in repl
Python
1 vote134 views1 answer
Code that works if variable has a value python
Python
0 vote155 views1 answer
Don't Get Volunteered!
Python