ndarray as the core data structure in NumPy - Time & Space Complexity

Time Complexity: ndarray as the core data structure

O(n)

Understanding Time Complexity

We want to understand how the time to work with numpy's ndarray changes as the data size grows.

How does the number of operations grow when we do common tasks on ndarrays?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import numpy as np

n = 10  # example size
arr = np.arange(n)  # create an array of size n
result = arr * 2   # multiply each element by 2

This code creates an ndarray of size n and multiplies every element by 2.

Identify Repeating Operations

Look for repeated work done as the array size grows.

• Primary operation: Multiplying each element in the array by 2.
• How many times: Once for each element, so n times.

How Execution Grows With Input

As the array size n grows, the number of multiplications grows the same way.

Input Size (n)	Approx. Operations
10	10 multiplications
100	100 multiplications
1000	1000 multiplications

Pattern observation: The operations increase directly with n, so doubling n doubles the work.

Final Time Complexity

Time Complexity: O(n)

This means the time to multiply all elements grows linearly with the number of elements.

Common Mistake

[X] Wrong: "Since numpy uses fast C code, the operation is constant time regardless of size."

[OK] Correct: Even though numpy is fast, it still must touch each element once, so time grows with n.

Interview Connect

Understanding how ndarray operations scale helps you explain performance in data tasks clearly and confidently.

Self-Check

"What if we replaced element-wise multiplication with a matrix multiplication of two n x n arrays? How would the time complexity change?"

Practice

1.

What is the main purpose of a numpy ndarray?

easy

A. To display images on screen

B. To create text documents

C. To store numbers in one or more dimensions for fast math operations

D. To connect to databases

Solution

1. Step 1: Understand what ndarray stores

   ndarray is designed to hold numbers arranged in rows and columns or more dimensions.

2. Step 2: Recognize its use for math

   It allows fast math operations on many numbers at once, unlike regular Python lists.

3. Final Answer:

   To store numbers in one or more dimensions for fast math operations -> Option C

4. Quick Check:

   ndarray = fast number storage [OK]

Hint: ndarray = fast multi-dimensional number storage [OK]

Common Mistakes:

• Thinking ndarray stores text or images
• Confusing ndarray with database tools
• Assuming ndarray is for display only

2.

Which of the following is the correct way to create a 1D ndarray from a Python list [1, 2, 3]?

import numpy as np
# Which line is correct?

easy

A. arr = np.array([1, 2, 3])

B. arr = np.ndarray([1, 2, 3])

C. arr = np.array(1, 2, 3)

D. arr = np.array{1, 2, 3}

Solution

1. Step 1: Recall correct syntax for np.array()

   np.array() takes a list or nested lists inside parentheses.

2. Step 2: Check each option

   arr = np.array([1, 2, 3]) uses correct syntax with square brackets inside parentheses. Others have wrong function calls or braces.

3. Final Answer:

   arr = np.array([1, 2, 3]) -> Option A

4. Quick Check:

   np.array(list) syntax = correct [OK]

Hint: Use np.array() with list inside parentheses [OK]

Common Mistakes:

• Using np.ndarray() directly without parameters
• Passing multiple arguments instead of one list
• Using curly braces {} instead of []

3.

What will be the output of the following code?

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

medium

A. (4,)

B. (2, 2)

C. (1, 4)

D. (2,)

Solution

1. Step 1: Understand the array shape

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

2. Step 2: Check print output

   arr.shape returns a tuple with dimensions, so output is (2, 2).

3. Final Answer:

   (2, 2) -> Option B

4. Quick Check:

   2 rows and 2 columns = (2, 2) [OK]

Hint: Shape shows rows and columns as tuple [OK]

Common Mistakes:

• Confusing shape with total elements
• Expecting a list instead of tuple
• Mixing up rows and columns count

4.

Find the error in this code snippet:

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

medium

A. SyntaxError due to wrong print statement

B. No error, prints 3

C. TypeError because arr is not a list

D. IndexError because index 3 is out of bounds

Solution

1. Step 1: Check array length and indexing

   arr has elements at indices 0, 1, 2 only. Index 3 is out of range.

2. Step 2: Understand error type

   Accessing arr[3] causes IndexError because index is invalid.

3. Final Answer:

   IndexError because index 3 is out of bounds -> Option D

4. Quick Check:

   Index out of range = IndexError [OK]

Hint: Check array length before indexing [OK]

Common Mistakes:

• Thinking arr[3] returns last element
• Confusing IndexError with SyntaxError
• Assuming numpy arrays allow out-of-range access

5.

You have a nested list [[1, 2], [3, 4], [5, 6]]. How do you create a numpy ndarray and then get the sum of all elements?

import numpy as np
nested_list = [[1, 2], [3, 4], [5, 6]]
# Fill in the blanks
arr = np.array(nested_list)
total = _____
print(total)

hard

A. arr.sum()

B. sum(arr)

C. arr.add()

D. np.sum(arr, axis=1)

Solution

1. Step 1: Create ndarray from nested list

   np.array(nested_list) creates a 2D array with shape (3, 2).

2. Step 2: Use ndarray method to sum all elements

   arr.sum() returns sum of all elements in the array.

3. Final Answer:

   arr.sum() -> Option A

4. Quick Check:

   Use arr.sum() for total sum [OK]

Hint: Use arr.sum() to sum all elements [OK]

Common Mistakes:

• Using Python sum() which sums rows, not elements
• Using arr.add() which does not exist
• Using np.sum with axis=1 sums rows, not total