/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 vote316 views1 answer
Django create object from dict
Python
0 vote348 views1 answer
Pretty size python
Python
0 vote311 views2 answers
Connect two mathod to the same button in pyq5
Python
0 vote371 views1 answer
I want to add a new column to the DataFrame containing only the month of the measurement
Python
0 vote357 views1 answer
Try python import
Python
1 vote316 views2 answers
Convert 2d list to 1d python
Python
0 vote349 views1 answer
Python Hewwo wowwd
Python
1 vote384 views2 answers
Operators in python
Python
44 votes317 views3 answers
Python add to start of list
Python
0 vote335 views1 answer
Pandas read chunk of csv
Python