/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 vote245 views1 answer
UnicodeEncodeError: 'latin-1' codec can't encode character '\u2013' in position 83: ordinal not in range(256)
Python
-1 vote278 views2 answers
Python print 2d array as table
Python
1 vote283 views1 answer
How to custom page not found in django
Python Django
0 vote235 views1 answer
\n vs \r python
Python
18 votes208 views1 answer
Dataframe tolist python
Python
0 vote256 views1 answer
Trends in yearly data python
Python
2 votes230 views1 answer
Use of == python
Python
0 vote250 views1 answer
Python hello world jenkins
Python
0 vote248 views1 answer
Python [:-1]
Python
0 vote241 views1 answer
Dynamic array logic in python use
Python