Categorical data type optimization in Data Analysis Python - Step-by-Step Execution

Concept Flow - Categorical data type optimization

Load DataFrame with object columns

↓

Check unique values count

↓

Convert suitable columns to 'category'

↓

Compare memory usage before and after

↓

Use optimized DataFrame for analysis

This flow shows how to convert object columns with repeated values into categorical type to save memory and speed up analysis.

Execution Sample

Data Analysis Python

import pandas as pd

# Create sample data
s = pd.Series(['apple', 'banana', 'apple', 'orange', 'banana', 'banana'])

# Convert to category
d = s.astype('category')

# Show memory usage
print(s.memory_usage(), d.memory_usage())

This code creates a series of fruit names, converts it to a categorical type, and compares memory usage.

Execution Table

Step	Action	Input/Variable	Result/Output
1	Create Series s	['apple', 'banana', 'apple', 'orange', 'banana', 'banana']	s: object dtype, 6 elements
2	Check s.memory_usage()	s	Memory usage: 176 bytes
3	Convert s to category	s.astype('category')	d: categorical dtype, 6 elements
4	Check d.memory_usage()	d	Memory usage: 136 bytes
5	Compare memory usage	s.memory_usage() vs d.memory_usage()	Categorical uses less memory
6	Use d for analysis	d	Optimized data for faster operations

💡 Conversion done, memory usage reduced, ready for optimized analysis

Variable Tracker

Variable	Start	After Conversion	Final
s	Series of strings (object)	Same (unchanged)	Series of strings (object)
d	Not defined	Series of categorical type	Series of categorical type

Key Moments - 3 Insights

Why does converting to 'category' reduce memory usage?

Because categorical stores unique values once and uses integer codes for data, as shown in execution_table step 4 where memory usage drops.

Can all object columns be converted to categorical?

No, only columns with repeated values and limited unique entries benefit; otherwise, it may not save memory or speed.

Does converting to categorical change the data values?

No, the visible data stays the same; only internal storage changes, as seen in variable_tracker where values remain consistent.

Visual Quiz - 3 Questions

Look at the execution_table, what is the memory usage of the original Series s at step 2?

A 136 bytes

B 176 bytes

C 6 bytes

D 200 bytes

💡 Hint

Check the 'Result/Output' column at step 2 in the execution_table.

At which step does the Series get converted to categorical type?

A Step 1

B Step 5

C Step 3

D Step 6

💡 Hint

Look for the action mentioning 'Convert s to category' in the execution_table.

If the Series had all unique values, what would likely happen to memory usage after conversion?

A Memory usage would stay about the same or increase

B Memory usage would become zero

C Memory usage would decrease significantly

D Conversion would fail

💡 Hint

Recall key_moments about when categorical conversion is beneficial.

Concept Snapshot

Categorical data type optimization:
- Use pd.Series.astype('category') to convert object columns
- Saves memory by storing unique values once
- Speeds up operations on repeated values
- Best for columns with few unique entries
- Check memory usage before and after conversion

Full Transcript

We start with a Series of strings representing fruits. Initially, it uses more memory because each string is stored fully. By converting this Series to a categorical type, pandas stores each unique fruit once and uses integer codes for the data. This reduces memory usage, as shown by comparing memory usage before and after conversion. Not all columns benefit; only those with repeated values and limited unique entries. The data values remain the same visually, only the internal storage changes. This optimization helps speed up data analysis and reduces memory load.