Understanding array memory layout in NumPy - Hands-On Activity

Understanding array memory layout

📖 Scenario: You are working with images stored as arrays. Understanding how these arrays are stored in memory helps you work faster and avoid mistakes.

🎯 Goal: You will create a NumPy array, check its memory layout flags, and print whether it is stored in row-major (C-contiguous) or column-major (Fortran-contiguous) order.

📋 What You'll Learn

Create a 2D NumPy array with exact values

Check the array's memory layout flags

Print if the array is C-contiguous or Fortran-contiguous

💡 Why This Matters

🌍 Real World

Understanding array memory layout helps when working with large datasets or images, making your code faster and more efficient.

💼 Career

Data scientists and engineers often optimize code by knowing how data is stored in memory, especially when using libraries like NumPy.

1

Create a 2D NumPy array

Import NumPy as np and create a 2D NumPy array called arr with these exact values: [[1, 2, 3], [4, 5, 6]].

NumPy

# Import numpy and create the array arr
# Your code here

Need a hint?

Use np.array() to create the array with the given nested list.

2

Check the memory layout flags

Create a variable called flags that stores the memory layout flags of arr using arr.flags.

NumPy

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
# Store the memory layout flags of arr in flags
# Your code here

Need a hint?

Use the flags attribute of the array to get memory layout info.

3

Determine if array is C-contiguous or Fortran-contiguous

Create a variable called layout. Use an if statement to set layout to the string 'C-contiguous' if flags.c_contiguous is True. Otherwise, set layout to 'Fortran-contiguous' if flags.f_contiguous is True. If neither, set layout to 'Non-contiguous'.

NumPy

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
flags = arr.flags
# Use if-else to set layout based on flags.c_contiguous and flags.f_contiguous
# Your code here

Need a hint?

Check flags.c_contiguous first, then flags.f_contiguous, else default.

4

Print the memory layout

Print the string "Memory layout of arr:" followed by the value of layout.

NumPy

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
flags = arr.flags
if flags.c_contiguous:
    layout = 'C-contiguous'
elif flags.f_contiguous:
    layout = 'Fortran-contiguous'
else:
    layout = 'Non-contiguous'
# Print the memory layout of arr
# Your code here

Need a hint?

Use print("Memory layout of arr:", layout) to show the result.