/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 vote141 views2 answers
Run all jupyter notebooks in project folder
Python
0 vote134 views1 answer
Add 2 number
Python
0 vote232 views1 answer
Resource punkt not found. Please use the NLTK Downloader to obtain the resource:
Python
0 vote146 views1 answer
Dictionary convert 2 lists into a dictionary, use zip()
Python
0 vote129 views1 answer
Download viper for python ubutunu
Python
0 vote138 views1 answer
Finns = False
Python
0 vote151 views1 answer
ImportError: cannot import name 'dedent' from 'matplotlib.cbook'
Python
0 vote156 views1 answer
Fetch row where column is missing pandas
Python
1 vote177 views1 answer
Pig latin translator python
Python
1 vote148 views1 answer
The month before python dateime
Python