/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 vote188 views1 answer
Python previous answer
Python
-2 votes172 views5 answers
Convert python to java
Python
0 vote198 views2 answers
Pandas replace non numeric values with 0?
Python
0 vote187 views1 answer
Can only concatenate str (not "ImageFieldFile") to str
Python
1 vote206 views1 answer
Python remove one element from numpy array
Python
10 votes186 views1 answer
Pandas dataframe creation column names
Python
13 votes184 views5 answers
Group by pandas count
Python
0 vote182 views1 answer
Get data from s3 bucket python
Python
4 votes149 views1 answer
Python comma separated string to list
Python
0 vote187 views1 answer
How to change dtype object to int
Python