Why advanced indexing matters in NumPy - See It in Action

Why advanced indexing matters

📖 Scenario: Imagine you have a small photo album represented as a grid of pixel brightness values. You want to pick specific pixels to analyze or change, but they are scattered in different rows and columns. Using normal indexing can be slow or complicated. Advanced indexing helps you pick exactly the pixels you want quickly and easily.

🎯 Goal: You will create a small 2D array of pixel brightness values, select specific pixels using advanced indexing, and print the selected pixels. This shows how advanced indexing helps pick scattered data points efficiently.

📋 What You'll Learn

Create a 2D numpy array called pixels with the exact values [[10, 20, 30], [40, 50, 60], [70, 80, 90]]

Create two numpy arrays called row_indices and col_indices with the exact values [0, 2, 1] and [1, 0, 2] respectively

Use advanced indexing with row_indices and col_indices to select specific pixels from pixels and store in selected_pixels

Print the selected_pixels array

💡 Why This Matters

🌍 Real World

Advanced indexing is useful in image processing, data analysis, and scientific computing where you need to pick specific scattered data points quickly.

💼 Career

Data scientists and analysts often use advanced indexing to manipulate and analyze large datasets efficiently, saving time and computing resources.

1

Create the pixel brightness array

Create a 2D numpy array called pixels with these exact values: [[10, 20, 30], [40, 50, 60], [70, 80, 90]]

NumPy

import numpy as np
# Create the 2D numpy array called pixels with the given values
# Your code here

Hint

Use np.array() and pass the list of lists exactly as shown.

2

Create index arrays for rows and columns

Create two numpy arrays called row_indices and col_indices with these exact values: [0, 2, 1] and [1, 0, 2] respectively

NumPy

import numpy as np
pixels = np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]])
# Create row_indices and col_indices numpy arrays with the given values
# Your code here

Hint

Use np.array() to create both arrays with the exact values.

3

Select pixels using advanced indexing

Use advanced indexing with row_indices and col_indices to select pixels from pixels and store the result in selected_pixels

NumPy

import numpy as np
pixels = np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]])
row_indices = np.array([0, 2, 1])
col_indices = np.array([1, 0, 2])
# Use advanced indexing to select pixels
# Your code here

Hint

Use pixels[row_indices, col_indices] to pick the scattered pixels.

4

Print the selected pixels

Print the selected_pixels array to see the chosen pixel brightness values

NumPy

import numpy as np
pixels = np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]])
row_indices = np.array([0, 2, 1])
col_indices = np.array([1, 0, 2])
selected_pixels = pixels[row_indices, col_indices]
# Print the selected_pixels array
# Your code here

Hint

Use print(selected_pixels) to display the result.

Practice

1. What is the main benefit of using advanced indexing in NumPy arrays?

easy

A. It allows selecting multiple specific elements easily and efficiently.

B. It automatically sorts the array elements.

C. It converts arrays into lists.

D. It changes the data type of the array.

Solution

1. Step 1: Understand what advanced indexing does

   Advanced indexing lets you pick many specific elements from an array using lists or arrays of indices.

2. Step 2: Compare options with this understanding

   Only It allows selecting multiple specific elements easily and efficiently. describes this benefit correctly; others describe unrelated features.

3. Final Answer:

   It allows selecting multiple specific elements easily and efficiently. -> Option A

4. Quick Check:

   Advanced indexing = easy multi-element selection [OK]

Hint: Advanced indexing picks many elements at once [OK]

Common Mistakes:

• Thinking it sorts or changes data types
• Confusing with basic slicing
• Assuming it converts arrays to lists

2. Which of the following is the correct syntax for advanced indexing to select elements at positions 0, 2, and 4 from a NumPy array arr?

easy

A. arr(0, 2, 4)

B. arr[[0, 2, 4]]

C. arr[0:2:4]

D. arr{0, 2, 4}

Solution

1. Step 1: Recall advanced indexing syntax

   Advanced indexing uses square brackets with a list or array of indices inside, like arr[[0, 2, 4]].

2. Step 2: Check each option

   arr[[0, 2, 4]] uses correct syntax. Options A, C, and D use invalid syntax for indexing.

3. Final Answer:

   arr[[0, 2, 4]] -> Option B

4. Quick Check:

   Use double brackets with list for advanced indexing [OK]

Hint: Use double brackets with list for advanced indexing [OK]

Common Mistakes:

• Using parentheses instead of brackets
• Using slicing syntax incorrectly
• Using curly braces which are invalid

3. What is the output of the following code?

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

medium

A. [1 3]

B. [10 30]

C. [20 40]

D. [20 30 40]

Solution

1. Step 1: Understand the array and indices

   The array is [10, 20, 30, 40, 50]. The indices list is [1, 3].

2. Step 2: Apply advanced indexing

   arr[indices] selects elements at positions 1 and 3, which are 20 and 40.

3. Final Answer:

   [20 40] -> Option C

4. Quick Check:

   arr[[1,3]] = [20 40] [OK]

Hint: Index list picks elements at those positions [OK]

Common Mistakes:

• Confusing indices with values
• Expecting a slice output
• Printing indices instead of array elements

4. The following code throws an error. What is the cause?

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

medium

A. Index 5 is out of bounds for the array.

B. The indices list must be a NumPy array.

C. Advanced indexing cannot use lists.

D. The array must be 2D for advanced indexing.

Solution

1. Step 1: Check the array size and indices

   The array has 5 elements with indices 0 to 4. The indices list includes 5, which is invalid.

2. Step 2: Understand the error cause

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

3. Final Answer:

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

4. Quick Check:

   Index must be within array size [OK]

Hint: Check if all indices are within array length [OK]

Common Mistakes:

• Thinking list type causes error
• Assuming advanced indexing needs 2D arrays
• Ignoring out-of-range indices

5. Given a 2D NumPy array arr = np.array([[1,2,3],[4,5,6],[7,8,9]]), how can you use advanced indexing to select elements 2, 6, and 7 from the array?

hard

A. arr[0:3, 1:3]

B. arr[[1,2,3], [2,3,1]]

C. arr[[0,1,2]][[1,2,0]]

D. arr[[0,1,2], [1,2,0]]

Solution

1. Step 1: Identify positions of elements 2, 6, and 7

   Element 2 is at (0,1), 6 at (1,2), and 7 at (2,0).

2. Step 2: Use advanced indexing with row and column arrays

   Using arr[[0,1,2], [1,2,0]] selects these elements by pairing rows and columns.

3. Final Answer:

   arr[[0,1,2], [1,2,0]] -> Option D

4. Quick Check:

   Pair row and column indices to select elements [OK]

Hint: Use paired row and column index arrays [OK]

Common Mistakes:

• Using invalid indices out of range
• Trying to slice instead of advanced indexing
• Misaligning row and column index arrays