Fancy indexing with integer arrays in NumPy

Introduction

Fancy indexing lets you pick specific items from data using lists of positions. It helps you get or change many values at once easily.

You want to select multiple specific rows from a table by their row numbers.

You need to update certain elements in a list or array based on their positions.

You want to reorder data by giving a new order of indexes.

You want to extract scattered values from a big dataset quickly.

Syntax

NumPy

import numpy as np

# Create an array
array = np.array([10, 20, 30, 40, 50])

# Use a list or array of integers to pick elements
indexes = [1, 3, 4]
selected_elements = array[indexes]

The indexes can be a Python list or a NumPy array of integers.

Fancy indexing returns a new array with the selected elements.

Examples

Selects elements at positions 0, 2, and 4 from the array.

NumPy

import numpy as np

array = np.array([5, 10, 15, 20, 25])
indexes = [0, 2, 4]
print(array[indexes])

Empty index list returns an empty array (edge case: no elements selected).

NumPy

import numpy as np

array = np.array([5, 10, 15, 20, 25])
indexes = []
print(array[indexes])

Single element array and selecting that one element.

NumPy

import numpy as np

array = np.array([5])
indexes = [0]
print(array[indexes])

Selects elements at the end and beginning, showing order can be changed.

NumPy

import numpy as np

array = np.array([5, 10, 15, 20, 25])
indexes = [4, 0]
print(array[indexes])

Sample Program

This program shows how to pick elements from an array using a list of indexes, then change those elements in the original array.

NumPy

import numpy as np

# Create an array of numbers
numbers = np.array([100, 200, 300, 400, 500])
print("Original array:", numbers)

# Choose indexes to select
selected_indexes = [1, 3, 4]

# Use fancy indexing to get elements at these positions
selected_numbers = numbers[selected_indexes]
print("Selected elements at positions 1, 3, 4:", selected_numbers)

# Change elements at these positions
numbers[selected_indexes] = [210, 410, 510]
print("Array after updating selected positions:", numbers)

Important Notes

Time complexity is O(k) where k is the number of indexes selected.

Space complexity is O(k) for the new array created by fancy indexing.

A common mistake is using indexes out of range, which causes an error.

Use fancy indexing when you want to select or update multiple specific elements at once. For continuous slices, normal slicing is simpler and faster.

Summary

Fancy indexing lets you pick or change many elements by giving their positions in a list.

It works with lists or arrays of integers as indexes.

It creates a new array of selected elements or updates the original array if assigned.

Practice

1. What does fancy indexing with integer arrays in NumPy allow you to do?

easy

A. Create a new array filled with zeros

B. Sort an array in ascending order

C. Select multiple elements from an array using a list of positions

D. Change the data type of an array

Solution

1. Step 1: Understand fancy indexing

   Fancy indexing uses a list or array of integers to pick elements by their positions.

2. Step 2: Identify the correct operation

   It selects multiple elements from the original array based on given indices.

3. Final Answer:

   Select multiple elements from an array using a list of positions -> Option C

4. Quick Check:

   Fancy indexing = select elements by positions [OK]

Hint: Fancy indexing picks elements by position list [OK]

Common Mistakes:

• Confusing fancy indexing with sorting
• Thinking it changes data type
• Assuming it creates arrays of zeros

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

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 fancy indexing syntax

   Fancy indexing uses double square brackets with a list or array of indices inside.

2. Step 2: Match syntax to options

   Only arr[[0, 2, 4]] uses correct double brackets and list format.

3. Final Answer:

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

4. Quick Check:

   Double brackets with list = correct fancy indexing [OK]

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

Common Mistakes:

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

3. What is the output of the following code?

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

medium

A. [20 40 50]

B. [10 30 40]

C. [20 30 40]

D. [50 40 20]

Solution

1. Step 1: Understand the array and indices

   Original array is [10, 20, 30, 40, 50]. Indices are [1, 3, 4].

2. Step 2: Select elements at given indices

   Elements at positions 1, 3, 4 are 20, 40, 50 respectively.

3. Final Answer:

   [20 40 50] -> Option A

4. Quick Check:

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

Hint: Match indices to array positions to get output [OK]

Common Mistakes:

• Mixing up indices and values
• Using wrong positions from array
• Confusing slicing with fancy indexing

4. Identify the error in the following code snippet:

import numpy as np
arr = np.array([5, 10, 15, 20])
indices = [1, 4]
print(arr[indices])

medium

A. IndexError because index 4 is out of bounds

B. SyntaxError due to wrong brackets

C. TypeError because indices is a list

D. No error, prints [10 20]

Solution

1. Step 1: Check array size and indices

   Array length is 4, valid indices are 0 to 3. Index 4 is invalid.

2. Step 2: Understand error type

   Accessing index 4 causes IndexError: index out of bounds.

3. Final Answer:

   IndexError because index 4 is out of bounds -> Option A

4. Quick Check:

   Index 4 invalid for length 4 array [OK]

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

Common Mistakes:

• Assuming list indices are always valid
• Confusing syntax errors with index errors
• Thinking list indices cause type errors

5. Given a 2D NumPy array arr = np.array([[1,2,3],[4,5,6],[7,8,9]]), how can you use fancy indexing to select elements at positions (0,2), (1,0), and (2,1) and create a new 1D array with these values?

hard

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

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

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

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

Solution

1. Step 1: Understand 2D fancy indexing

   To select elements at (row, col) positions, use two index arrays: one for rows, one for columns.

2. Step 2: Match given positions to index arrays

   Positions are (0,2), (1,0), (2,1), so rows = [0,1,2], cols = [2,0,1].

3. Step 3: Use fancy indexing syntax

   Use arr[[0,1,2], [2,0,1]] to get elements [3,4,8].

4. Final Answer:

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

5. Quick Check:

   Rows and cols arrays match positions [OK]

Hint: Use two index arrays: rows and columns [OK]

Common Mistakes:

• Mixing row and column indices order
• Using chained indexing instead of fancy indexing
• Using wrong index arrays for positions