Memory usage analysis in Data Analysis Python - Time & Space Complexity

Time Complexity: Time complexity analysis

O(n * m)

Understanding Time Complexity

When we analyze time complexity, we want to understand how the runtime grows as the data size grows.

We ask: How does the work increase as the input size grows?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

def calculate_mean(df):
    means = {}
    for col in df.columns:
        means[col] = df[col].mean()
    return means

# df is a DataFrame with n rows and m columns

This code calculates the average value for each column in a data table.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Looping over each column and computing the mean of all rows in that column.
• How many times: The loop runs once for each column (m times), and inside each, the mean function processes all rows (n times).

How Execution Grows With Input

As the number of rows (n) or columns (m) grows, the work grows too.

Input Size (n rows, m columns)	Approx. Operations
10 rows, 5 columns	About 50 operations (10*5)
100 rows, 5 columns	About 500 operations (100*5)
1000 rows, 10 columns	About 10,000 operations (1000*10)

Pattern observation: The total work grows by multiplying rows and columns, so doubling either doubles the work.

Final Time Complexity

Time Complexity: O(n * m)

This means the time to calculate all means grows proportionally with the number of rows times the number of columns.

Common Mistake

[X] Wrong: "Calculating the mean for each column is just O(m) because we loop over columns only."

[OK] Correct: Each mean calculation looks at all rows, so the work inside the loop depends on n, making total work depend on both n and m.

Interview Connect

Understanding how data size affects processing time is key in data science. This skill helps you explain and improve data handling in real projects.

Self-Check

"What if we used a built-in function that calculates means for all columns at once? How would the time complexity change?"

Practice

1. What does the memory_usage(deep=True) method in pandas do?

easy

A. It counts the number of rows in the DataFrame.

B. It deletes unused memory from the DataFrame.

C. It compresses the DataFrame to use less memory.

D. It shows detailed memory usage of each column including object types.

Solution

1. Step 1: Understand memory_usage method

   The memory_usage method reports memory used by each column in a DataFrame.

2. Step 2: Effect of deep=True parameter

   Using deep=True makes it include memory used by object types like strings, giving detailed info.

3. Final Answer:

   It shows detailed memory usage of each column including object types. -> Option D

4. Quick Check:

   memory_usage(deep=True) = detailed memory info [OK]

Hint: Remember deep=True reveals full memory use including strings [OK]

Common Mistakes:

• Thinking it deletes or compresses data
• Confusing it with counting rows
• Ignoring deep=True effect

2. Which of the following is the correct syntax to get memory usage of a pandas DataFrame df including deep analysis?

easy

A. memory_usage(df, deep=True)

B. df.memoryusage(deep=True)

C. df.memory_usage(deep=True)

D. df.get_memory_usage(deep=True)

Solution

1. Step 1: Recall pandas method name

   The correct method is memory_usage with underscore, called on DataFrame.

2. Step 2: Check method call syntax

   It is called as df.memory_usage(deep=True). Other options have wrong method names or syntax.

3. Final Answer:

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

4. Quick Check:

   Correct method name and call = df.memory_usage(deep=True) [OK]

Hint: Use exact method name: memory_usage with underscore [OK]

Common Mistakes:

• Misspelling method as memoryusage
• Using function style instead of method
• Adding get_ prefix incorrectly

3. Given the following code, what is the output?

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

medium

A. Error because deep=True is invalid

B. Index memory + int64 column + object column memory sizes

C. Total memory size as a single number

D. Only memory size of numeric columns

Solution

1. Step 1: Understand DataFrame columns

   Column 'A' is integers, 'B' is strings (object type), and index is default integer index.

2. Step 2: memory_usage(deep=True) output

   This method returns a Series showing memory used by index and each column separately, including deep memory for object column 'B'.

3. Final Answer:

   Index memory + int64 column + object column memory sizes -> Option B

4. Quick Check:

   memory_usage(deep=True) lists memory per column [OK]

Hint: memory_usage(deep=True) shows memory per column including index [OK]

Common Mistakes:

• Expecting a single total number output
• Ignoring index memory
• Thinking deep=True causes error

4. What is wrong with this code snippet?

import pandas as pd
df = pd.DataFrame({'col': ['a', 'b', 'c']})
print(df.memory_usage(deep=False))

medium

A. It does not show full memory usage of string data.

B. The method memory_usage does not exist.

C. deep=False causes a syntax error.

D. DataFrame must have numeric columns to use memory_usage.

Solution

1. Step 1: Check memory_usage with deep=False

   By default or with deep=False, memory_usage does not count full memory used by object columns like strings.

2. Step 2: Effect on output

   This means the reported memory for 'col' will be underestimated, missing actual string memory.

3. Final Answer:

   It does not show full memory usage of string data. -> Option A

4. Quick Check:

   deep=False skips detailed object memory [OK]

Hint: Use deep=True to get full string memory usage [OK]

Common Mistakes:

• Thinking memory_usage method is missing
• Assuming deep=False causes error
• Believing numeric columns are required

5. You have a large DataFrame with many string columns. You want to reduce memory usage without losing data. Which approach helps most based on memory usage analysis?

hard

A. Convert string columns to categorical type after checking memory usage.

B. Drop all string columns to save memory.

C. Use memory_usage(deep=False) to estimate memory and ignore strings.

D. Convert all columns to float type regardless of original data.

Solution

1. Step 1: Analyze memory usage of string columns

   Using memory_usage(deep=True) shows string columns use a lot of memory.

2. Step 2: Convert strings to categorical

   Changing string columns to categorical reduces memory by storing unique values once.

3. Step 3: Avoid data loss

   Converting to categorical keeps data but uses less memory, unlike dropping or wrong type conversion.

4. Final Answer:

   Convert string columns to categorical type after checking memory usage. -> Option A

5. Quick Check:

   Convert strings to categorical for memory saving [OK]

Hint: Convert strings to categorical to save memory without losing data [OK]

Common Mistakes:

• Dropping important columns blindly
• Ignoring deep=True memory insights
• Converting all columns to float causing errors