Memory usage analysis in Pandas

Introduction

Memory usage analysis helps you understand how much computer memory your data uses. This is important to keep your programs fast and avoid crashes.

When working with large datasets and you want to check if your computer can handle them.

Before saving or sharing data, to know how big the file might be in memory.

When optimizing your code to use less memory and run faster.

To compare memory use before and after changing data types or cleaning data.

Syntax

Pandas

DataFrame.memory_usage(index=True, deep=False)

index: If True, includes memory used by the index.

deep: If True, gives a more accurate memory use by inspecting object types deeply.

Examples

Shows memory used by each column and the index (default index=True).

Pandas

df.memory_usage()

Shows memory used by columns only, excluding the index.

Pandas

df.memory_usage(index=False)

Gives a detailed memory usage including the actual size of objects like strings.

Pandas

df.memory_usage(deep=True)

Sample Program

This code creates a small table with names, ages, and cities. It then shows memory used by each column and the index, including detailed size for strings. Finally, it sums all to show total memory used.

Pandas

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Charlie'],
        'age': [25, 30, 35],
        'city': ['New York', 'Los Angeles', 'Chicago']}

df = pd.DataFrame(data)

# Check memory usage including index
mem_usage = df.memory_usage(deep=True)
print(mem_usage)

# Total memory used by DataFrame
total_mem = mem_usage.sum()
print(f"Total memory usage: {total_mem} bytes")

Important Notes

Using deep=True is helpful for object columns like strings to get accurate memory use.

Memory usage is shown in bytes, which is a small unit of computer memory.

Checking memory helps you decide if you need to reduce data size or change data types.

Summary

Memory usage analysis tells you how much memory your data takes.

Use memory_usage() to see memory per column and index.

Use deep=True for more accurate memory info on objects.

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