Why memory management matters in NumPy - Challenge Your Understanding

Challenge - 5 Problems

❓ Predict Output

intermediate

Understanding numpy array memory usage

What is the output of the following code showing the memory size of a numpy array?

NumPy

import numpy as np
arr = np.arange(1000)
print(arr.nbytes)

A 1000

B 4000

C 8000

D 16000

💡 Hint

Each element in the array is an integer of 8 bytes (int64) by default.

The array has 1000 integers, each 8 bytes (int64), so total memory is 8000 bytes.

❓ data_output

intermediate

Effect of data type on memory usage

Which option shows the correct memory size in bytes for a numpy array of 1000 elements with dtype float64?

NumPy

import numpy as np
arr = np.arange(1000, dtype=np.float64)
print(arr.nbytes)

A 32000

B 16000

C 4000

D 8000

💡 Hint

float64 uses 8 bytes per element.

1000 elements * 8 bytes each = 8000 bytes.

🧠 Conceptual

advanced

Why memory management is important in data science

Why is managing memory important when working with large datasets in numpy?

A To reduce the chance of running out of memory and slowing down computations

B Because numpy automatically frees memory so manual management is not needed

C To make the code run slower and use more CPU

D Because memory management only matters for small datasets

💡 Hint

Think about what happens if your computer runs out of memory.

Large datasets can use a lot of memory. Managing memory helps avoid crashes and slowdowns.

🔧 Debug

advanced

Identifying memory inefficiency in numpy code

Which option shows the code that wastes the most memory when creating large arrays?

NumPy

import numpy as np

# Option A
arr_a = np.arange(1000000, dtype=np.int32)

# Option B
arr_b = np.arange(1000000, dtype=np.int64)

# Option C
arr_c = np.arange(1000000, dtype=np.float64)

# Option D
arr_d = np.arange(1000000, dtype=np.int16)

A arr_a uses 4MB, arr_b uses 8MB, arr_c uses 8MB, arr_d uses 2MB; all use the same memory

B arr_a uses 4MB, arr_b uses 8MB, arr_c uses 8MB, arr_d uses 2MB; arr_b and arr_c waste the most memory

C arr_a uses 4MB, arr_b uses 8MB, arr_c uses 8MB, arr_d uses 2MB; arr_a wastes the most memory

D arr_a uses 4MB, arr_b uses 8MB, arr_c uses 8MB, arr_d uses 2MB; arr_d wastes the most memory

💡 Hint

Check the size in bytes of each data type.

int64 and float64 use 8 bytes per element, which is more than int32 and int16.

🚀 Application

expert

Optimizing memory usage for large numpy arrays

You have a numpy array of 1 million integers ranging from 0 to 255. Which dtype should you choose to minimize memory usage without losing data?

A np.uint8

B np.int32

C np.float32

D np.int64

💡 Hint

Consider the range of values and the size of each data type.

Since values are between 0 and 255, unsigned 8-bit integer (uint8) uses 1 byte per element and fits all values.