Why indexing matters in NumPy - Performance Analysis

Time Complexity: Why indexing matters

O(k)

Understanding Time Complexity

We want to see how fast numpy operations run when we pick parts of data using indexing.

How does the way we select data affect the time it takes?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import numpy as np

arr = np.arange(1000000)

# Access a single element
single = arr[500000]

# Access a slice
slice_ = arr[100000:200000]

# Access with a boolean mask
mask = arr % 2 == 0
filtered = arr[mask]

This code shows three ways to get data from a large array: one element, a slice, and a filtered subset.

Identify Repeating Operations

Look at what repeats when we pick data.

• Primary operation: Accessing elements by index or condition.
• How many times: For single element, once; for slice, many times equal to slice size; for mask, once per element to check condition.

How Execution Grows With Input

How does time grow when array size grows?

Input Size (n)	Approx. Operations
10	Single: 1, Slice: up to 10, Mask: 10
100	Single: 1, Slice: up to 100, Mask: 100
1000	Single: 1, Slice: up to 1000, Mask: 1000

Picking one element stays fast no matter size. Picking slices or filtered data grows with how much we take or check.

Final Time Complexity

Time Complexity: O(k) where k is the number of elements accessed or checked

This means time grows with how many elements you pick or test, not the total array size if you pick just one.

Common Mistake

[X] Wrong: "Accessing any element in numpy always takes the same time regardless of how many elements we get."

[OK] Correct: Accessing one element is fast, but getting many elements or filtering means more work and more time.

Interview Connect

Knowing how indexing affects speed helps you write faster code and explain your choices clearly in real projects.

Self-Check

"What if we used fancy indexing with a list of random indices? How would the time complexity change?"

Practice

1. What does indexing in NumPy allow you to do?

easy

A. Create new arrays from scratch

B. Select specific elements or slices from an array

C. Automatically sort the array elements

D. Convert arrays into lists

Solution

1. Step 1: Understand the purpose of indexing

   Indexing is used to access specific parts of an array, like single elements or slices.

2. Step 2: Compare with other options

   Creating new arrays, sorting, or converting are different operations, not indexing.

3. Final Answer:

   Select specific elements or slices from an array -> Option B

4. Quick Check:

   Indexing = selecting parts of array [OK]

Hint: Indexing means picking parts of data quickly [OK]

Common Mistakes:

• Confusing indexing with array creation
• Thinking indexing sorts data
• Mixing indexing with data type conversion

2. Which of the following is the correct syntax to get the first element of a NumPy array arr?

easy

A. arr.first()

B. arr(0)

C. arr[0]

D. arr{0}

Solution

1. Step 1: Recall NumPy indexing syntax

   NumPy arrays use square brackets with zero-based indices to access elements.

2. Step 2: Check each option

   Only arr[0] uses correct square brackets and zero index; others are invalid syntax or methods.

3. Final Answer:

   arr[0] -> Option C

4. Quick Check:

   First element = arr[0] [OK]

Hint: Use square brackets and zero-based index [OK]

Common Mistakes:

• Using parentheses instead of brackets
• Using curly braces which are invalid
• Assuming methods like .first() exist

3. What is the output of the following code?

import numpy as np
arr = np.array([10, 20, 30, 40, 50])
print(arr[1:4])

medium

A. [20 30 40]

B. [10 20 30 40]

C. [20 30 40 50]

D. [30 40 50]

Solution

1. Step 1: Understand slicing syntax

   The slice arr[1:4] selects elements from index 1 up to but not including 4.

2. Step 2: Identify elements at these indices

   Indices 1, 2, 3 correspond to values 20, 30, 40.

3. Final Answer:

   [20 30 40] -> Option A

4. Quick Check:

   Slice 1:4 excludes index 4 [OK]

Hint: Slice excludes end index, so 1:4 means 1,2,3 [OK]

Common Mistakes:

• Including the end index element
• Confusing zero-based indexing
• Printing the whole array by mistake

4. What is wrong with this code snippet?

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

medium

A. Index 5 is out of bounds for the array

B. The array is not defined properly

C. The print statement syntax is incorrect

D. The array should be a list, not a NumPy array

Solution

1. Step 1: Check array length and indexing

   The array has 5 elements with indices 0 to 4; index 5 is invalid.

2. Step 2: Identify error cause

   Accessing arr[5] causes an IndexError because it is out of bounds.

3. Final Answer:

   Index 5 is out of bounds for the array -> Option A

4. Quick Check:

   Index must be less than array length [OK]

Hint: Array indices go from 0 to length-1 [OK]

Common Mistakes:

• Trying to access index equal to array length
• Confusing array length with max index
• Assuming Python allows out-of-range indexing

5. Given a 2D NumPy array arr = np.array([[1,2,3],[4,5,6],[7,8,9]]), which indexing expression will select the second column as a 1D array?

hard

A. arr[1, :]

B. arr[:, 2]

C. arr[1]

D. arr[:, 1]

Solution

1. Step 1: Understand 2D array indexing

   In 2D arrays, arr[:, 1] means all rows, column index 1 (second column).

2. Step 2: Check other options

   arr[1, :] selects second row, arr[1] is same as second row, arr[:, 2] selects third column.

3. Final Answer:

   arr[:, 1] -> Option D

4. Quick Check:

   All rows, column 1 = second column [OK]

Hint: Use colon for all rows, index for column [OK]

Common Mistakes:

• Mixing row and column indices
• Using single index for 2D array
• Selecting wrong column index