Handling missing values in ML Python

Introduction

Missing values can cause errors or wrong results in machine learning. Handling them helps models learn better and make good predictions.

When your dataset has empty or unknown entries in some columns.

When you want to prepare data before training a model.

When you want to avoid errors caused by missing data during analysis.

When you want to keep as much data as possible without losing rows.

When you want to fill missing data with reasonable guesses.

Syntax

ML Python

from sklearn.impute import SimpleImputer

imputer = SimpleImputer(strategy='mean')
X_imputed = imputer.fit_transform(X)

SimpleImputer replaces missing values with a chosen strategy like mean, median, or most frequent value.

You must fit the imputer on training data, then transform both training and test data.

Examples

Replace missing values with the mean of each column.

ML Python

imputer = SimpleImputer(strategy='mean')
X_imputed = imputer.fit_transform(X)

Replace missing values with the median of each column.

ML Python

imputer = SimpleImputer(strategy='median')
X_imputed = imputer.fit_transform(X)

Replace missing values with the most common value in each column.

ML Python

imputer = SimpleImputer(strategy='most_frequent')
X_imputed = imputer.fit_transform(X)

Replace missing values with zero.

ML Python

imputer = SimpleImputer(strategy='constant', fill_value=0)
X_imputed = imputer.fit_transform(X)

Sample Model

This program shows how to replace missing values with the mean of each column using SimpleImputer.

ML Python

import numpy as np
from sklearn.impute import SimpleImputer

# Sample data with missing values (np.nan)
X = np.array([[1, 2], [np.nan, 3], [7, np.nan], [np.nan, np.nan]])

# Create imputer to fill missing values with mean
imputer = SimpleImputer(strategy='mean')

# Fit imputer on data and transform
X_imputed = imputer.fit_transform(X)

print("Original data:\n", X)
print("\nData after imputing missing values with mean:\n", X_imputed)

Important Notes

Always fit the imputer only on training data to avoid data leakage.

Imputation strategies depend on the data type and distribution.

For categorical data, use 'most_frequent' or 'constant' strategies.

Summary

Missing values can cause problems in machine learning models.

SimpleImputer helps fill missing values with mean, median, or other strategies.

Always fit on training data and transform both training and test sets.

Practice

1. What is the main reason to handle missing values before training a machine learning model?

easy

A. Missing values are automatically ignored by all models

B. Missing values can cause errors or reduce model accuracy

C. Missing values improve model training speed

D. Missing values increase the size of the dataset

Solution

1. Step 1: Understand the impact of missing values

   Missing values can cause errors or confuse the model, leading to poor predictions.

2. Step 2: Recognize model requirements

   Most machine learning models require complete data without missing values to work properly.

3. Final Answer:

   Missing values can cause errors or reduce model accuracy -> Option B

4. Quick Check:

   Missing values harm model accuracy = D [OK]

Hint: Missing data breaks models, so fix it first [OK]

Common Mistakes:

• Thinking missing values speed up training
• Assuming models ignore missing data automatically
• Believing missing values increase dataset size

2. Which of the following is the correct way to import and use SimpleImputer from scikit-learn to fill missing values with the mean?

easy

A. from sklearn.impute import SimpleImputer\nimputer = SimpleImputer(method='mean')

B. import SimpleImputer from sklearn.impute\nimputer = SimpleImputer('mean')

C. from sklearn.imputer import SimpleImputer\nimputer = SimpleImputer(strategy=mean)

D. from sklearn.impute import SimpleImputer\nimputer = SimpleImputer(strategy='mean')

Solution

1. Step 1: Check import syntax

   The correct import is 'from sklearn.impute import SimpleImputer'.

2. Step 2: Check SimpleImputer initialization

   The strategy parameter must be a string like 'mean', so strategy='mean' is correct.

3. Final Answer:

   from sklearn.impute import SimpleImputer\nimputer = SimpleImputer(strategy='mean') -> Option D

4. Quick Check:

   Correct import and strategy syntax = A [OK]

Hint: Use strategy='mean' and correct import syntax [OK]

Common Mistakes:

• Using 'method' instead of 'strategy'
• Not quoting 'mean' as a string
• Wrong import statement syntax

3. Given the following code, what will be the output of print(X_imputed)?

import numpy as np
from sklearn.impute import SimpleImputer

X = np.array([[1, 2], [np.nan, 3], [7, np.nan]])
imputer = SimpleImputer(strategy='mean')
imputer.fit(X)
X_imputed = imputer.transform(X)
print(X_imputed)

medium

A. [[1. 2.]
[4. 3.]
[7. 2.5]]

B. [[1. 2.]
[nan 3.]
[7. nan]]

C. [[1. 2.]
[4. 3.]
[7. nan]]

D. Error due to missing values

Solution

1. Step 1: Calculate mean ignoring missing values

   Column 0 mean = (1 + 7) / 2 = 4; Column 1 mean = (2 + 3) / 2 = 2.5

2. Step 2: Replace missing values with column means

   Replace np.nan in row 2, col 0 with 4; replace np.nan in row 3, col 1 with 2.5

3. Final Answer:

   [[1. 2.] [4. 3.] [7. 2.5]] -> Option A

4. Quick Check:

   Missing values replaced by column means = B [OK]

Hint: Mean replaces NaN with average of column values [OK]

Common Mistakes:

• Not calculating mean correctly
• Leaving NaNs unchanged
• Expecting an error instead of imputation

4. Identify the error in the following code snippet for imputing missing values:

from sklearn.impute import SimpleImputer

X_train = [[1, 2], [None, 3], [7, None]]
imputer = SimpleImputer(strategy='median')
imputer.transform(X_train)

medium

A. Using null instead of np.nan causes error

B. Incorrect strategy name 'median'

C. Missing fit() call before transform()

D. SimpleImputer cannot handle 2D lists

Solution

1. Step 1: Check method sequence

   SimpleImputer requires fit() before transform() to learn statistics from data.

2. Step 2: Verify data and strategy

   Strategy 'median' is valid; None is accepted as missing; 2D list is acceptable.

3. Final Answer:

   Missing fit() call before transform() -> Option C

4. Quick Check:

   fit() must precede transform() = A [OK]

Hint: Always fit before transform with SimpleImputer [OK]

Common Mistakes:

• Calling transform without fit
• Confusing strategy names
• Thinking None is invalid missing value

5. You have a training set with missing values and a test set with missing values in the same columns. What is the best practice to handle missing values before training and evaluating a model?

hard

A. Fit the imputer on training data, then transform both training and test data

B. Fit the imputer separately on training and test data

C. Fill missing values in test data with zeros, training data with mean

D. Remove all rows with missing values from both datasets

Solution

1. Step 1: Understand data leakage risk

   Fitting imputer separately on test data leaks information from test set, causing biased evaluation.

2. Step 2: Apply consistent imputation

   Fit imputer only on training data to learn statistics, then apply same transformation to test data.

3. Final Answer:

   Fit the imputer on training data, then transform both training and test data -> Option A

4. Quick Check:

   Fit on train, transform both = C [OK]

Hint: Fit imputer on train only, then transform test [OK]

Common Mistakes:

• Fitting imputer separately on test data
• Filling test missing values with zeros arbitrarily
• Dropping too much data unnecessarily