/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

2 votes141 views1 answer
Python loop list of dictionaries
Python
1 vote167 views1 answer
Python convert string to sentence case
Python
0 vote217 views1 answer
Demploy django in vps
Python Django
0 vote194 views1 answer
Applies a function to all elements of this RDD.
Python
0 vote210 views1 answer
Python search a string in another string get last result
Python
0 vote171 views1 answer
Drop rows with null date in pandas
Python
1 vote144 views1 answer
Input on discord.py d
Python
1 vote234 views2 answers
Np.sort descending
Python
0 vote186 views1 answer
[ WARN:0] global C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-nxx381if\opencv\modules\videoio\src\cap_msmf.cpp (435) `anonymous-namespace'
Python
0 vote210 views1 answer
Username__icontains in django
Python Django