Practical demonstration of SimpleImputer

simpleimputer_basic.py

# import the required library
from sklearn.impute import SimpleImputer

# define imputer
imputer = SimpleImputer(strategy='mean') # other options include: median, most_frequent, constant

# fit on the training dataset
imputer.fit(X_train)

# transform the training dataset (don't forget to transform the test dataset as well)
X_train_clean = pd.DataFrame(imputer.transform(X_train), columns = X_train.columns, index = X_train.index)