Array shapes and dimensions in Data Analysis Python - Time & Space Complexity

Time Complexity: Array shapes and dimensions

O(n x m)

Understanding Time Complexity

When working with arrays, knowing how their size and shape affect processing time is important.

We want to understand how the time to handle arrays grows as their dimensions change.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import numpy as np

def sum_all_elements(arr):
    total = 0
    for i in range(arr.shape[0]):
        for j in range(arr.shape[1]):
            total += arr[i, j]
    return total

# Example usage:
# arr = np.array([[1, 2], [3, 4]])
# sum_all_elements(arr)

This code sums all elements in a 2D array by visiting each element one by one.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Nested loops visiting each element in the 2D array.
• How many times: Outer loop runs for number of rows, inner loop runs for number of columns, so total visits equal rows x columns.

How Execution Grows With Input

As the array grows in rows and columns, the total operations grow by multiplying these dimensions.

Input Size (rows x columns)	Approx. Operations
10 x 10	100
100 x 100	10,000
1000 x 1000	1,000,000

Pattern observation: Doubling rows and columns roughly quadruples the operations because total elements multiply.

Final Time Complexity

Time Complexity: O(n * m)

This means the time grows proportionally to the total number of elements in the array, which is rows times columns.

Common Mistake

[X] Wrong: "The time depends only on the number of rows or columns, not both."

[OK] Correct: Because the code visits every element, both rows and columns matter. Ignoring one dimension underestimates the work done.

Interview Connect

Understanding how array dimensions affect processing time helps you explain efficiency clearly in interviews and shows you can reason about data size impact.

Self-Check

"What if we changed the array to 3 dimensions? How would the time complexity change?"

Practice

1. What does the shape of a NumPy array represent?

easy

A. The number of elements along each dimension

B. The total number of elements in the array

C. The data type of the elements

D. The memory size of the array

Solution

1. Step 1: Understand the meaning of shape

   The shape of an array is a tuple that shows how many elements are in each dimension.

2. Step 2: Differentiate shape from other properties

   Shape is not the total count, data type, or memory size, but the size per dimension.

3. Final Answer:

   The number of elements along each dimension -> Option A

4. Quick Check:

   Shape = elements per dimension [OK]

Hint: Shape = count of elements per axis [OK]

Common Mistakes:

• Confusing shape with total number of elements
• Mixing shape with data type
• Thinking shape is memory size

2. Which of the following is the correct way to get the shape of a NumPy array arr?

easy

A. arr.shape

B. arr.shape()

C. shape(arr)

D. arr.get_shape()

Solution

1. Step 1: Recall NumPy shape syntax

   In NumPy, shape is an attribute, not a method, so no parentheses are used.

2. Step 2: Check each option

   arr.shape uses arr.shape correctly; others use incorrect syntax or functions.

3. Final Answer:

   arr.shape -> Option A

4. Quick Check:

   Use attribute, no parentheses [OK]

Hint: Use arr.shape without parentheses [OK]

Common Mistakes:

• Adding parentheses after shape
• Using non-existent functions like get_shape()
• Trying to call shape as a function

3. Given the code:

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape)

What is the output?

medium

A. (3, 2)

B. (6,)

C. (2, 3)

D. (2,)

Solution

1. Step 1: Analyze the array structure

   The array has 2 rows and 3 columns, so shape is (2, 3).

2. Step 2: Confirm shape output

   Printing arr.shape returns the tuple (2, 3).

3. Final Answer:

   (2, 3) -> Option C

4. Quick Check:

   Rows=2, Columns=3 -> (2, 3) [OK]

Hint: Count rows then columns for shape [OK]

Common Mistakes:

• Swapping rows and columns in shape
• Thinking shape is total elements
• Confusing shape with array size

4. What is wrong with this code snippet?

import numpy as np
arr = np.array([1, 2, 3])
print(arr.shape())

medium

A. print statement syntax is incorrect

B. arr is not a NumPy array

C. The array is empty

D. shape is an attribute, not a method; parentheses cause error

Solution

1. Step 1: Identify the error source

   The code calls arr.shape() with parentheses, but shape is an attribute.

2. Step 2: Understand attribute vs method

   Attributes do not use parentheses; calling with () causes a TypeError.

3. Final Answer:

   shape is an attribute, not a method; parentheses cause error -> Option D

4. Quick Check:

   Attributes no parentheses [OK]

Hint: Remember: shape is attribute, no () [OK]

Common Mistakes:

• Using parentheses with shape
• Assuming array is empty causes error
• Misreading print syntax

5. You have a 3D NumPy array with shape (4, 3, 2). Which of the following operations will correctly reshape it into a 2D array with 4 rows and 6 columns?

hard

A. arr.reshape(12, 2)

B. arr.reshape(4, 6)

C. arr.reshape(6, 4)

D. arr.reshape(2, 12)

Solution

1. Step 1: Calculate total elements

   Original shape (4, 3, 2) means 4*3*2 = 24 elements total.

2. Step 2: Determine new shape dimensions

   New shape (4, 6) means 4*6 = 24 elements, matching total elements.

3. Step 3: Verify other options

   arr.reshape(12, 2), arr.reshape(6, 4), and arr.reshape(2, 12) do not produce 4 rows and 6 columns.

4. Final Answer:

   arr.reshape(4, 6) -> Option B

5. Quick Check:

   4*6=24 matches total elements [OK]

Hint: New shape must keep total elements same [OK]

Common Mistakes:

• Mixing up rows and columns in reshape
• Choosing shapes with wrong total elements
• Forgetting reshape must keep total size