/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 vote238 views1 answer
Raise keyerror(key) from none os.environ
Python
1 vote214 views1 answer
Python list remove at index
Python
0 vote218 views1 answer
Inspectdb django
Python
0 vote229 views1 answer
Create python package ros 2
Python
1 vote247 views1 answer
Tuple negative indexing in python
Python
2 votes219 views2 answers
Python get location of script
Python
4 votes308 views2 answers
Read binary file python
Python
1 vote239 views1 answer
Python for loop even numbers
Python
0 vote224 views1 answer
Python program to switch first and second characters in a string
Python
1 vote229 views1 answer
Unlimited keyword arguments python
Python