/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

-1 vote287 views1 answer
Jupyter notebook show full dataframe cell
Python
0 vote301 views1 answer
Python timeit function return value
Python
4 votes214 views1 answer
Datetime has no attribute now
Python
2 votes253 views3 answers
Fibonacci formula python
Python
3 votes347 views3 answers
Find elements in selenium
Python
0 vote296 views1 answer
Fetch firestore indexes
Python
9 votes296 views1 answer
Bee movie script
Python
0 vote280 views1 answer
No package python37 available.
Python
0 vote312 views1 answer
Python show charracter code
Python
1 vote295 views1 answer
How to check if python has been added to path
Python