ColumnTransformer for mixed types in ML Python

Introduction

We use ColumnTransformer to apply different data changes to different columns in one step. This helps when data has numbers and words mixed together.

You have a table with some columns as numbers and others as words.

You want to change numbers by scaling and words by turning them into numbers.

You want to prepare data quickly before teaching a computer to learn.

You want to keep your data changes organized and easy to repeat.

Syntax

ML Python

from sklearn.compose import ColumnTransformer
from sklearn.preprocessing import StandardScaler, OneHotEncoder

transformer = ColumnTransformer(
    transformers=[
        ('num', StandardScaler(), ['num_column1', 'num_column2']),
        ('cat', OneHotEncoder(), ['cat_column1', 'cat_column2'])
    ]
)

Each transformer has a name, a method, and the columns it changes.

Transformers run in parallel and combine results automatically.

Examples

Scale 'age' and 'income' columns, and one-hot encode 'city' column.

ML Python

ColumnTransformer(
    transformers=[
        ('scale', StandardScaler(), ['age', 'income']),
        ('encode', OneHotEncoder(), ['city'])
    ]
)

Scale 'height' and encode 'color' with safe handling of new categories.

ML Python

ColumnTransformer(
    transformers=[
        ('num', StandardScaler(), ['height']),
        ('cat', OneHotEncoder(handle_unknown='ignore'), ['color'])
    ]
)

Sample Model

This example shows how to use ColumnTransformer to scale numbers and encode categories before training a logistic regression model. It splits data, trains, predicts, and shows accuracy.

ML Python

from sklearn.compose import ColumnTransformer
from sklearn.preprocessing import StandardScaler, OneHotEncoder
from sklearn.linear_model import LogisticRegression
from sklearn.pipeline import Pipeline
from sklearn.model_selection import train_test_split
import numpy as np
import pandas as pd

# Sample data with mixed types
data = pd.DataFrame({
    'age': [25, 32, 47, 51, 62],
    'income': [50000, 60000, 80000, 72000, 90000],
    'city': ['New York', 'Paris', 'Paris', 'London', 'New York'],
    'target': [0, 1, 0, 1, 0]
})

X = data.drop('target', axis=1)
y = data['target']

# Define ColumnTransformer
preprocessor = ColumnTransformer(
    transformers=[
        ('num', StandardScaler(), ['age', 'income']),
        ('cat', OneHotEncoder(), ['city'])
    ]
)

# Create a pipeline with preprocessing and model
model = Pipeline(steps=[('preprocessor', preprocessor),
                        ('classifier', LogisticRegression())])

# Split data
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=42)

# Train model
model.fit(X_train, y_train)

# Predict
predictions = model.predict(X_test)

# Print results
print('Predictions:', predictions)
print('Test labels:', y_test.values)
print(f'Test accuracy: {model.score(X_test, y_test):.2f}')

Important Notes

ColumnTransformer keeps your data changes clear and easy to manage.

Always match column names exactly when specifying columns.

Use pipelines to combine preprocessing and model training smoothly.

Summary

ColumnTransformer lets you change different columns in different ways at once.

It is useful when your data has both numbers and words.

Use it with pipelines to prepare data and train models easily.

Practice

1. What is the main purpose of using ColumnTransformer in machine learning?

easy

A. To train multiple models on the same dataset

B. To apply different preprocessing steps to different columns in a dataset

C. To visualize data distributions

D. To split data into training and testing sets

Solution

1. Step 1: Understand the role of ColumnTransformer

   ColumnTransformer allows applying different transformations to different columns, such as scaling numbers and encoding text.

2. Step 2: Compare with other options

   Training models, visualizing data, or splitting data are different tasks not handled by ColumnTransformer.

3. Final Answer:

   To apply different preprocessing steps to different columns in a dataset -> Option B

4. Quick Check:

   ColumnTransformer = Different preprocessing per column [OK]

Hint: Think: Different columns, different treatments [OK]

Common Mistakes:

• Confusing ColumnTransformer with model training
• Thinking it splits data instead of transforming
• Assuming it visualizes data

2. Which of the following is the correct way to import ColumnTransformer from scikit-learn?

easy

A. from sklearn.feature_extraction import ColumnTransformer

B. from sklearn.preprocessing import ColumnTransformer

C. from sklearn.pipeline import ColumnTransformer

D. from sklearn.compose import ColumnTransformer

Solution

1. Step 1: Recall the module for ColumnTransformer

   ColumnTransformer is part of the compose module in scikit-learn.

2. Step 2: Verify other options

   Preprocessing, pipeline, and feature_extraction modules do not contain ColumnTransformer.

3. Final Answer:

   from sklearn.compose import ColumnTransformer -> Option D

4. Quick Check:

   ColumnTransformer is in compose module [OK]

Hint: Remember: compose module for combining transformers [OK]

Common Mistakes:

• Importing from preprocessing instead of compose
• Confusing pipeline with compose
• Trying to import from feature_extraction

3. Given the code below, what will be the output of print(transformed_data)?

from sklearn.compose import ColumnTransformer
from sklearn.preprocessing import StandardScaler, OneHotEncoder
import numpy as np

X = np.array([[1, 'red'], [2, 'blue'], [3, 'green']])

ct = ColumnTransformer([
    ('num', StandardScaler(), [0]),
    ('cat', OneHotEncoder(), [1])
])

transformed_data = ct.fit_transform(X)
print(transformed_data)

medium

A. A numpy array with scaled numbers and one-hot encoded colors

B. A list of original values without changes

C. An error because StandardScaler cannot handle strings

D. A numpy array with only scaled numbers

Solution

1. Step 1: Understand ColumnTransformer setup

   Column 0 (numbers) is scaled; column 1 (colors) is one-hot encoded.

2. Step 2: Predict output structure

   Output is a numpy array combining scaled numeric values and one-hot encoded categorical values.

3. Final Answer:

   A numpy array with scaled numbers and one-hot encoded colors -> Option A

4. Quick Check:

   Mixed types transformed correctly = scaled + one-hot [OK]

Hint: Remember: ColumnTransformer applies each transformer to specified columns [OK]

Common Mistakes:

• Expecting original data without transformation
• Thinking StandardScaler will fail on mixed data
• Ignoring one-hot encoding effect

4. What is wrong with the following code snippet using ColumnTransformer?

from sklearn.compose import ColumnTransformer
from sklearn.preprocessing import StandardScaler, OneHotEncoder
import numpy as np

X = np.array([[1, 'red'], [2, 'blue'], [3, 'green']])

ct = ColumnTransformer([
    ('num', StandardScaler(), [0, 1]),
    ('cat', OneHotEncoder(), [1])
])

transformed_data = ct.fit_transform(X)

medium

A. StandardScaler is applied to a string column causing an error

B. OneHotEncoder is applied to a numeric column causing an error

C. ColumnTransformer requires numeric data only

D. No error, code runs fine

Solution

1. Step 1: Check columns assigned to StandardScaler

   StandardScaler is applied to columns 0 and 1, but column 1 contains strings.

2. Step 2: Understand why this causes an error

   StandardScaler cannot process string data, so this will raise a type error.

3. Final Answer:

   StandardScaler is applied to a string column causing an error -> Option A

4. Quick Check:

   Scaler on strings = error [OK]

Hint: Scaler only works on numeric columns [OK]

Common Mistakes:

• Applying scaler to categorical columns
• Assuming ColumnTransformer auto-detects types
• Ignoring column indices in transformer

5. You have a dataset with numeric columns ['age', 'income'] and categorical columns ['city', 'gender']. You want to scale numeric columns and one-hot encode categorical columns using ColumnTransformer. Which code snippet correctly sets this up?

hard

A. ColumnTransformer([('num', OneHotEncoder(), ['age', 'income']), ('cat', StandardScaler(), ['city', 'gender'])])

B. ColumnTransformer([('num', StandardScaler(), ['city', 'gender']), ('cat', OneHotEncoder(), ['age', 'income'])])

C. ColumnTransformer([('num', StandardScaler(), ['age', 'income']), ('cat', OneHotEncoder(), ['city', 'gender'])])

D. ColumnTransformer([('num', StandardScaler(), ['age']), ('cat', OneHotEncoder(), ['income', 'city', 'gender'])])

Solution

1. Step 1: Identify correct transformers for each column type

   Numeric columns should be scaled with StandardScaler; categorical columns should be one-hot encoded.

2. Step 2: Match columns to transformers correctly

   ColumnTransformer([('num', StandardScaler(), ['age', 'income']), ('cat', OneHotEncoder(), ['city', 'gender'])]) assigns numeric columns to StandardScaler and categorical columns to OneHotEncoder correctly.

3. Final Answer:

   ColumnTransformer([('num', StandardScaler(), ['age', 'income']), ('cat', OneHotEncoder(), ['city', 'gender'])]) -> Option C

4. Quick Check:

   Numeric scaled + categorical one-hot = ColumnTransformer([('num', StandardScaler(), ['age', 'income']), ('cat', OneHotEncoder(), ['city', 'gender'])]) [OK]

Hint: Match scaler to numbers, encoder to categories [OK]

Common Mistakes:

• Swapping transformers between numeric and categorical
• Mixing columns in wrong transformer
• Leaving out columns