/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

3 votes273 views2 answers
Django radio button
Python Django
-1 vote254 views3 answers
Print 5x5 2d list python
Python
13 votes321 views3 answers
Django form example
Python Django
1 vote284 views1 answer
Sklearn accuracy
Python
11 votes295 views4 answers
Python remove empty values from list
Python
0 vote257 views1 answer
Python tkinter askopenfile
Python
7 votes293 views3 answers
Python split string by specific word
Python
0 vote246 views1 answer
Can paypal be hacked by email
Python
59 votes271 views6 answers
Python replace character in string
Python
44 votes269 views6 answers
Plt figsize
Python