/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 vote309 views1 answer
List of words from a string and filter them based on a secondary list
Python
0 vote366 views1 answer
Redirect if not logged in django
Python Django
0 vote297 views1 answer
Convert pb to tb with python
Python
2 votes313 views2 answers
Random.randint(0,20) + pyrthon
Python
0 vote305 views1 answer
How to input a full array in one input in python
Python
0 vote284 views1 answer
Rest plus
Python
0 vote325 views1 answer
Matplotlib create histogram edge color
Python
12 votes318 views2 answers
Python string to hex
Python
0 vote311 views1 answer
Python enumerate list with comprehension
Python
6 votes333 views4 answers
How to ask a question in python
Python