Categorical data type optimization in Data Analysis Python - Time & Space Complexity

Time Complexity: Categorical data type optimization

O(n)

Understanding Time Complexity

When working with large data tables, converting text columns to categorical types can speed up operations.

We want to know how the time to convert grows as the data size grows.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

def optimize_categorical(df, col):
    df[col] = df[col].astype('category')
    return df

# Example usage:
# df = optimize_categorical(df, 'city')

This code changes a column's data type to categorical to save memory and speed up some operations.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Scanning the entire column to find unique values and assign codes.
• How many times: Once over all rows in the column.

How Execution Grows With Input

The time to convert grows roughly in direct proportion to the number of rows.

Input Size (n)	Approx. Operations
10	About 10 checks and assignments
100	About 100 checks and assignments
1000	About 1000 checks and assignments

Pattern observation: Doubling the rows roughly doubles the work done.

Final Time Complexity

Time Complexity: O(n)

This means the time to convert grows linearly with the number of rows in the column.

Common Mistake

[X] Wrong: "Converting to categorical is instant no matter the data size."

[OK] Correct: The process scans all rows to find unique values, so bigger data takes more time.

Interview Connect

Understanding how data type changes affect speed helps you work efficiently with big data in real projects.

Self-Check

"What if the column already has very few unique values? How would that affect the time complexity?"

Practice

1.

What is the main benefit of converting a DataFrame column to category type in pandas?

easy

A. It reduces memory usage by storing unique values once

B. It automatically sorts the column values alphabetically

C. It converts numeric values to strings

D. It removes missing values from the column

Solution

1. Step 1: Understand what categorical type does

   The categorical type stores unique values only once and uses codes internally, saving memory.

2. Step 2: Compare with other options

   Sorting, converting to strings, or removing missing values are not automatic effects of categorical conversion.

3. Final Answer:

   It reduces memory usage by storing unique values once -> Option A

4. Quick Check:

   Categorical type = Memory saving [OK]

Hint: Categorical stores unique values once to save memory [OK]

Common Mistakes:

• Thinking it sorts values automatically
• Confusing with string conversion
• Assuming it removes missing data

2.

Which of the following is the correct syntax to convert a pandas DataFrame column df['color'] to categorical type?

import pandas as pd
df = pd.DataFrame({'color': ['red', 'blue', 'red', 'green']})

easy

A. df['color'] = df.astype('category')

B. df['color'] = df['color'].astype('category')

C. df['color'] = pd.category(df['color'])

D. df['color'] = df['color'].to_category()

Solution

1. Step 1: Identify correct method to convert column type

   The astype('category') method converts a Series to categorical type.

2. Step 2: Check other options for errors

   df['color'] = df.astype('category') tries to convert whole DataFrame, not column. df['color'] = pd.category(df['color']) uses non-existent pd.category(). df['color'] = df['color'].to_category() uses wrong method name.

3. Final Answer:

   df['color'] = df['color'].astype('category') -> Option B

4. Quick Check:

   astype('category') = Correct syntax [OK]

Hint: Use astype('category') on the specific column [OK]

Common Mistakes:

• Applying astype to whole DataFrame instead of column
• Using non-existent pd.category() function
• Using wrong method like to_category()

3.

What will be the output of the following code?

import pandas as pd

df = pd.DataFrame({'fruit': ['apple', 'banana', 'apple', 'orange', 'banana']})
df['fruit'] = df['fruit'].astype('category')
print(df['fruit'].cat.categories)

medium

A. Index(['apple', 'banana', 'orange'], dtype='object')

B. ['apple', 'banana', 'apple', 'orange', 'banana']

C. ['apple', 'banana', 'orange', 'apple', 'banana']

D. Error: 'cat' attribute not found

Solution

1. Step 1: Convert column to categorical and check categories

   After conversion, cat.categories shows unique sorted categories.

2. Step 2: Identify unique values in sorted order

   Unique fruits are 'apple', 'banana', 'orange' in alphabetical order.

3. Final Answer:

   Index(['apple', 'banana', 'orange'], dtype='object') -> Option A

4. Quick Check:

   cat.categories = unique sorted values [OK]

Hint: cat.categories shows unique sorted categories [OK]

Common Mistakes:

• Expecting original list instead of unique categories
• Confusing categories with values
• Thinking cat attribute is missing

4.

Identify the error in this code snippet that tries to convert a column to categorical:

import pandas as pd
df = pd.DataFrame({'size': ['S', 'M', 'L', 'M']})
df['size'] = df['size'].astype(category)

medium

A. astype() cannot convert to categorical

B. DataFrame creation syntax is wrong

C. Missing quotes around 'category' in astype()

D. Column 'size' does not exist

Solution

1. Step 1: Check astype() argument

   The argument to astype() must be a string 'category', not a bare word.

2. Step 2: Verify other parts of code

   DataFrame creation and column name are correct; astype() supports 'category'.

3. Final Answer:

   Missing quotes around 'category' in astype() -> Option C

4. Quick Check:

   Quotes needed for dtype string [OK]

Hint: Always put quotes around dtype names in astype() [OK]

Common Mistakes:

• Forgetting quotes around 'category'
• Assuming astype() can't convert to category
• Mistaking column name errors

5.

You have a DataFrame with a column 'city' containing 1 million rows but only 5 unique city names. Which approach optimizes memory and speeds up grouping by city?

hard

A. Use df.groupby('city') without any conversion

B. Convert 'city' column to string type before grouping

C. Drop duplicates from 'city' column before grouping

D. Convert 'city' column to categorical type before grouping

Solution

1. Step 1: Understand memory and speed benefits of categorical

   Converting to categorical stores unique values once, reducing memory and speeding grouping.

2. Step 2: Evaluate other options

   String type uses more memory; dropping duplicates loses data; grouping without conversion is slower.

3. Final Answer:

   Convert 'city' column to categorical type before grouping -> Option D

4. Quick Check:

   Categorical = Memory + speed optimization [OK]

Hint: Convert to categorical before grouping large repeated data [OK]

Common Mistakes:

• Using string type wastes memory
• Dropping duplicates loses data
• Grouping without conversion is slower