Memory usage analysis in Pandas - Time & Space Complexity

Time Complexity: Memory usage analysis

O(n)

Understanding Time Complexity

We want to understand how the time needed to check memory usage grows as data size grows in pandas.

Specifically, how long does it take to measure memory use when the data gets bigger?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

n = 1000  # example value for n

data = pd.DataFrame({
    'A': [str(i) for i in range(n)],
    'B': [str(i) for i in range(n, 2*n)]
})

memory = data.memory_usage(deep=True)

This code creates a DataFrame with n rows and then checks how much memory it uses.

Identify Repeating Operations

• Primary operation: pandas checks memory for each column and each element inside if deep=True.
• How many times: It goes through all columns and all rows once to sum memory.

How Execution Grows With Input

As the number of rows (n) grows, the time to check memory grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	About 10 checks per column
100	About 100 checks per column
1000	About 1000 checks per column

Pattern observation: The time grows linearly as the data size grows.

Final Time Complexity

Time Complexity: O(n)

This means the time to measure memory grows in a straight line with the number of rows.

Common Mistake

[X] Wrong: "Checking memory usage is instant and does not depend on data size."

[OK] Correct: pandas must look at each element to calculate memory, so bigger data takes more time.

Interview Connect

Knowing how memory checks scale helps you understand performance when working with big data in pandas.

Self-Check

"What if we set deep=False in memory_usage()? How would the time complexity change?"

Practice

1. What does the memory_usage() function in pandas primarily show?

easy

A. The amount of memory used by each column and the index in a DataFrame

B. The total number of rows in a DataFrame

C. The data types of each column in a DataFrame

D. The number of missing values in each column

Solution

1. Step 1: Understand the purpose of memory_usage()

   This function reports memory consumption of each column and the index in a pandas DataFrame.

2. Step 2: Compare with other options

   Options A, B, and D describe different DataFrame properties unrelated to memory usage.

3. Final Answer:

   The amount of memory used by each column and the index in a DataFrame -> Option A

4. Quick Check:

   memory_usage() = memory per column and index [OK]

Hint: memory_usage() shows memory per column and index [OK]

Common Mistakes:

• Confusing memory_usage() with shape or info()
• Thinking it shows data types or missing values
• Ignoring that it includes index memory

2. Which of the following is the correct syntax to get detailed memory usage of a DataFrame df including object types?

easy

A. df.memory_usage()

B. df.memory_usage(True)

C. df.memory_usage(deep=True)

D. df.memory_usage(deep=False)

Solution

1. Step 1: Recall the parameter for detailed memory info

   The deep=True argument tells pandas to include full memory usage for object columns.

2. Step 2: Check syntax correctness

   df.memory_usage(deep=True) uses the correct keyword argument deep=True. df.memory_usage(True) uses positional argument incorrectly. Options A, B, and C do not provide detailed memory info.

3. Final Answer:

   df.memory_usage(deep=True) -> Option C

4. Quick Check:

   Use deep=True for detailed memory [OK]

Hint: Use deep=True to get full memory info on objects [OK]

Common Mistakes:

• Passing True without keyword (positional argument error)
• Not using deep=True and missing object memory
• Using deep=False which is default and less detailed

3. Given the following code:

import pandas as pd

df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': ['x', 'y', 'z']
})
print(df.memory_usage(deep=True))

What will be the output type of print(df.memory_usage(deep=True))?

medium

A. A pandas Series showing memory usage per column and index

B. A single integer showing total memory usage

C. A dictionary with column names as keys and memory usage as values

D. A DataFrame with memory usage details

Solution

1. Step 1: Understand the return type of memory_usage()

   The memory_usage() method returns a pandas Series with index as column names plus the index, and values as memory usage in bytes.

2. Step 2: Check the code output type

   Since deep=True is used, it includes detailed memory for object columns but still returns a Series, not a dict or DataFrame.

3. Final Answer:

   A pandas Series showing memory usage per column and index -> Option A

4. Quick Check:

   memory_usage() returns Series [OK]

Hint: memory_usage() returns Series with memory per column [OK]

Common Mistakes:

• Expecting total memory as single integer
• Thinking it returns a dictionary or DataFrame
• Confusing output type with other pandas methods

4. You run df.memory_usage(deep=True) but get an error: TypeError: memory_usage() got an unexpected keyword argument 'deep'. What is the likely cause?

medium

A. You forgot to import pandas as pd

B. You are using an older pandas version that does not support the 'deep' argument

C. You passed 'deep' as a positional argument instead of keyword

D. The DataFrame is empty

Solution

1. Step 1: Analyze the error message

   The error says 'unexpected keyword argument deep', meaning the function does not recognize 'deep'.

2. Step 2: Identify cause based on pandas version

   Older pandas versions did not support the 'deep' argument in memory_usage(). This is the most common cause.

3. Final Answer:

   You are using an older pandas version that does not support the 'deep' argument -> Option B

4. Quick Check:

   Old pandas versions lack deep argument [OK]

Hint: Check pandas version if 'deep' argument causes error [OK]

Common Mistakes:

• Assuming import error causes this
• Thinking positional argument fixes it
• Blaming empty DataFrame for this error

5. You have a large DataFrame with many object columns. You want to reduce memory usage by converting some columns to categorical type. Which approach helps you analyze memory before and after conversion?

hard

A. Use df.memory_usage() without deep argument before conversion

B. Use df.info() only after conversion

C. Use df.describe() before conversion

D. Use df.memory_usage(deep=True) before and after converting columns to 'category' dtype

Solution

1. Step 1: Understand memory usage measurement

   Using memory_usage(deep=True) gives detailed memory info including object columns, which is important before conversion.

2. Step 2: Compare memory before and after conversion

   By running this method before and after converting columns to 'category', you can see the memory savings clearly.

3. Final Answer:

   Use df.memory_usage(deep=True) before and after converting columns to 'category' dtype -> Option D

4. Quick Check:

   memory_usage(deep=True) shows detailed memory changes [OK]

Hint: Compare memory_usage(deep=True) before and after changes [OK]

Common Mistakes:

• Using info() which shows less detailed memory
• Using describe() which summarizes stats, not memory
• Using memory_usage() without deep misses object memory