/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 vote315 views1 answer
Best Python Free Tutorial
Python
0 vote334 views2 answers
Pandas sub columns
Python
0 vote346 views1 answer
Keras backend matrix multiplication
Python
6 votes393 views2 answers
Flask docker
Python Flask
0 vote316 views1 answer
Truthy falsy python
Python
0 vote332 views1 answer
Pandas turn counts into probability
Python
0 vote334 views1 answer
First_last6
Python
0 vote353 views1 answer
Get question mark from a word + python
Python
0 vote322 views1 answer
Why pytest return No ModuleError
Python
7 votes364 views3 answers
Convert uppercase to lowercase and vice versa in python
Python