/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 vote359 views1 answer
Purpose of migration folder in django
Python Django
2 votes323 views1 answer
Google scikit learn decision tree
Python
0 vote345 views1 answer
Unity python
Python
3 votes400 views1 answer
How to make downloadable file in flask
Python Flask
9 votes355 views3 answers
How to skip number in while loop python
Python
1 vote381 views1 answer
Ipython save session
Python
0 vote363 views1 answer
From threading import thread ImportError: cannot import name 'thread' from 'threading' (/usr/lib/python3.7/threading.py)
Python
1 vote372 views1 answer
Radiobuttons django
Python Django
1 vote334 views1 answer
Pandas plot xlabel
Python
0 vote371 views1 answer
Return the number of elements in this RDD.
Python