/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 vote364 views1 answer
Installing packages from requirements.txt file
Python
15 votes333 views3 answers
Remove punctuation python string library
Python
0 vote304 views1 answer
Def batting(balls,runs): points=runs/2; if runs>50: points=points+5; if runs>=100: points=points+10; strikerate=runs/balls; if strikerate>=80 and strikerate<=100: points=points+2; elif strikerate>100: points=points+4; print(points)
Python
0 vote349 views2 answers
Python suppress print output from function
Python
0 vote368 views1 answer
Clear variable jupyter notebook
Python
16 votes373 views6 answers
Print function python
Python
0 vote349 views1 answer
All the positions of a letter occurrences in a string python
Python
9 votes477 views2 answers
Flask how to run app
Python Flask
0 vote349 views1 answer
Arrow.get(null)
Python
0 vote367 views2 answers
Merge two sorted lists into one sorted list
Python