/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 vote316 views1 answer
Python pandas first and last element of column
Python
21 votes304 views4 answers
Floor function in python
Python
0 vote316 views1 answer
Import system in python
Python
0 vote325 views1 answer
Convert numpy array to tfrecord and back
Python
0 vote306 views2 answers
Django rest framework foreign key relation giving error in serializer
Python
1 vote331 views1 answer
Insert row in any position pandas dataframe
Python
0 vote343 views1 answer
Python pipe select where
Python
24 votes519 views6 answers
Static files in django
Python Django
0 vote359 views1 answer
Gets an existing SparkSession or, if there is no existing one, creates a new
Python
3 votes320 views2 answers
Python get line of exception
Python