/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

9 votes250 views1 answer
Sleep in python 3
Python
-1 vote221 views1 answer
Convert character to unicode python
Python
0 vote211 views4 answers
Python prevent print output
Python
14 votes206 views5 answers
Discord py get user by id
Python
0 vote189 views1 answer
Python program that gives the user a positive integer N, prints the number in the N position of the Fibonacci sequence
Python
0 vote207 views10 answers
Convert python to c++ online
Python
1 vote195 views3 answers
Link in embed discord.py
Python
0 vote245 views1 answer
Fiboacci in python
Python
0 vote209 views1 answer
Python capitalize first letter of string without changing the rest
Python
0 vote178 views1 answer
Iterrrows
Python