/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

7 votes182 views3 answers
Concat dataframes
Python
0 vote243 views1 answer
Python get majority of list
Python
0 vote206 views1 answer
Django less than and greater than
Python Django
14 votes241 views2 answers
How to change values in dataframe python
Python
0 vote169 views1 answer
Compress excel file in python
Python
0 vote189 views1 answer
Try: if '1' != 1: raise "someError" else: print("someError has not occurred") except "someError": print ("someError has occurred")
Python
6 votes170 views2 answers
Numpy from csv
Python
1 vote200 views2 answers
Valeur absolue python
Python
67 votes281 views6 answers
Deleting duplicates in list python
Python
0 vote178 views1 answer
"not equal to" python symbol
Python