/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 vote161 views4 answers
Dataframe to dictionary without index
Python
3 votes166 views2 answers
Different dataframe name with for loop
Python
1 vote221 views1 answer
How to add captcha in django forms
Python Django
1 vote170 views1 answer
Select rows which entries equals one of the values pandas
Python
25 votes168 views3 answers
Python warnings as error
Python
0 vote170 views1 answer
Add one to a column pands
Python
15 votes166 views3 answers
Python - gropuby based on 2 variabels
Python
1 vote153 views2 answers
Python split string on char
Python
0 vote153 views1 answer
Check if list elememnt in dataframe column
Python
0 vote206 views1 answer
Redirect if not logged in django
Python Django