Replacing NaNs in a Dataframe Column with Mean

Python

Uses the simple imputer in Sklearn to replace the NaNs in a dataframe column with the mean of the column.

 1|  from sklearn.impute import SimpleImputer
 2|  import numpy as np
 3|  
 4|  imputer = SimpleImputer(missing_values=np.nan, strategy='mean')
 5|  imputer.fit(df[column_name])
 6|  df[column_name] = imputer.transform(df[columns_name]))
Greg He
Sklearn | Data-cleaning | Numpy
Did you find this snippet useful?

Sign up for free to to add this to your code library

Remove Stop Words from Text in DataFrame Column
Python

Pandas | Nltk | Nlp

143
LightGBM Custom Loss Function
Python

LightGBM | Loss function

128
Tuning XGBoost Hyperparameters with Grid Search
Python

XGBoost | Tuning

123
How to Convert DataFrame Values Into Percentages
Python

Lambda | Percentages | Pipe

116
How to Scale Data Using Standard Scaler But Keep Column Names
Python

Scaler | Standard

87
LightGBM Hyperparameter Tuning with GridSearch
Python

Python | LightGBM | Hyperparameter tuning | Gridsearch

86
Calculating Root Mean Squared Error (RMSE) with Sklearn and Python
Python

Rmse

82
Dynamically Create Columns in Pandas Dataframe
Python

Pandas | Columns

66
How to Train a Catboost Classifier with GridSearch Hyperparameter Tuning
Python

Catboost | Hyperparameter tuning

65
Overwriting a Python Logging File Instead of Appending
Python

Logging

62