/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 vote205 views1 answer
Beautifulsoup find get value
Python
3 votes203 views2 answers
Count how many vowels in a string python
Python
20 votes215 views2 answers
Create a list of a certain length python
Python
10 votes204 views2 answers
Check if key exists in sesison python
Python
2 votes223 views1 answer
From Crypto.Cipher import AES ModuleNotFoundError: No module named 'Crypto'
Python
1 vote208 views2 answers
Build dataframe from dictionary
Python
0 vote226 views1 answer
Pandas correlation matrix between one column and all others
Python
1 vote191 views1 answer
Fastest sort python
Python
0 vote216 views1 answer
Python code to fetch all the files with txt extension from a folder
Python
0 vote214 views1 answer
Creating a Pandas Data Frame Series
Python