Single element access in NumPy - Practice Problems & Coding Challenges

Challenge - 5 Problems

❓ Predict Output

intermediate

Accessing a single element in a 2D NumPy array

What is the output of the following code?

NumPy

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

A 60

B [60]

C 50

D IndexError

💡 Hint

Remember that indexing starts at 0 and you access elements by row and column.

The element at row 1, column 2 is 60 because indexing starts at 0. So arr[1, 2] accesses the second row and third column.

❓ data_output

intermediate

Extracting a single element from a 3D NumPy array

Given the 3D array below, what is the value of arr[0, 1, 2]?

NumPy

import numpy as np
arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
print(arr[0, 1, 2])

A 9

B 5

C 6

D IndexError

💡 Hint

Think of the array as layers, rows, and columns. The first index selects the layer.

arr[0, 1, 2] selects the first layer (index 0), second row (index 1), and third column (index 2), which is 6.

🔧 Debug

advanced

Identify the error in single element access

What error will the following code raise?

NumPy

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

A IndexError: index 3 is out of bounds for axis 0 with size 3

B TypeError: list indices must be integers or slices

C ValueError: too many indices for array

D No error, output is 3

💡 Hint

Check the size of the array and the index used.

The array has indices 0, 1, 2. Accessing index 3 is out of bounds, so it raises an IndexError.

🚀 Application

advanced

Accessing a single element in a masked NumPy array

Given the masked array below, what is the output of masked_arr[1]?

NumPy

import numpy as np
import numpy.ma as ma
arr = np.array([10, 20, 30])
masked_arr = ma.masked_array(arr, mask=[False, True, False])
print(masked_arr[1])

A ValueError

B 20

C masked

D --

💡 Hint

Masked elements show as '--' when printed.

The element at index 1 is masked, so printing it shows '--' indicating a masked value.

🧠 Conceptual

expert

Understanding single element access with fancy indexing

What is the output of the following code?

NumPy

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

A TypeError

B 20

C IndexError

D [20]

💡 Hint

Remember that arr[[1]] returns a 1-element array, then [0] accesses the first element.

arr[[1]] returns an array with one element [20]. Accessing [0] extracts the scalar 20.

Practice

1. What does single element access in a numpy array allow you to do?

easy

A. Sort the entire array

B. Add new rows to the array

C. Read or change one specific value in the array

D. Calculate the sum of all elements

Solution

1. Step 1: Understand the concept of single element access

   Single element access means selecting one value from the array using its position.

2. Step 2: Identify what single element access can do

   It allows reading or changing that one value, not adding rows, sorting, or summing.

3. Final Answer:

   Read or change one specific value in the array -> Option C

4. Quick Check:

   Single element access = read/change one value [OK]

Hint: Single element access targets one value by position [OK]

Common Mistakes:

• Confusing single element access with adding or removing elements
• Thinking it can perform operations like sum or sort
• Mixing up element access with slicing

2. Which of the following is the correct syntax to access the element in the 2nd row and 3rd column of a numpy array named arr?

easy

A. arr[2, 3]

B. arr[1, 2]

C. arr(1, 2)

D. arr[2][3]

Solution

1. Step 1: Recall zero-based indexing in numpy

   Indexing starts at 0, so 2nd row is index 1, 3rd column is index 2.

2. Step 2: Check syntax for multi-dimensional access

   Use square brackets with comma-separated indexes: arr[1, 2]. Parentheses or double brackets are incorrect here.

3. Final Answer:

   arr[1, 2] -> Option B

4. Quick Check:

   Zero-based indexing + arr[row, col] = arr[1, 2] [OK]

Hint: Use zero-based indexes inside square brackets with commas [OK]

Common Mistakes:

• Using 1-based indexes instead of zero-based
• Using parentheses instead of square brackets
• Using double brackets like arr[2][3] which is less efficient

3. Given the numpy array:

import numpy as np
arr = np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]])
print(arr[2, 1])

What will be the output?

medium

A. 80

B. 70

C. 60

D. 50

Solution

1. Step 1: Identify the element at row 2, column 1

   Indexes are zero-based, so row 2 means third row: [70, 80, 90], column 1 means second element.

2. Step 2: Find the value at arr[2, 1]

   The value is 80.

3. Final Answer:

   80 -> Option A

4. Quick Check:

   arr[2,1] = 80 [OK]

Hint: Remember zero-based indexing: row 2 is third row [OK]

Common Mistakes:

• Confusing row and column indexes
• Using 1-based indexing
• Reading the wrong element from the array

4. What is wrong with this code snippet?

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

medium

A. TypeError because arr is not subscriptable

B. SyntaxError due to wrong brackets

C. It prints 3 correctly

D. IndexError because row index 2 is out of bounds

Solution

1. Step 1: Check array shape and valid indexes

   arr has shape (2, 2), so valid row indexes are 0 and 1 only.

2. Step 2: Identify the error from arr[2, 1]

   Row index 2 is out of bounds, causing IndexError.

3. Final Answer:

   IndexError because row index 2 is out of bounds -> Option D

4. Quick Check:

   IndexError from invalid row index [OK]

Hint: Check array shape before accessing indexes [OK]

Common Mistakes:

• Assuming indexes start at 1
• Confusing syntax errors with index errors
• Ignoring array dimensions

5. You have a 3D numpy array data with shape (4, 3, 2). How do you access the single element in the 4th block, 2nd row, and 1st column?

hard

A. data[3, 1, 0]

B. data[4, 2, 1]

C. data[3][1][1]

D. data[2, 1, 0]

Solution

1. Step 1: Understand zero-based indexing for 3D arrays

   4th block means index 3, 2nd row means index 1, 1st column means index 0.

2. Step 2: Write the correct index tuple

   Use data[3, 1, 0] to access the element.

3. Final Answer:

   data[3, 1, 0] -> Option A

4. Quick Check:

   3D array access = data[3,1,0] [OK]

Hint: Count from zero for each dimension in order [OK]

Common Mistakes:

• Using 1-based indexes instead of zero-based
• Mixing up the order of indexes
• Using invalid indexes beyond shape