Why indexing matters in NumPy

Introduction

Indexing helps you quickly find and use specific parts of your data. It saves time and makes your work easier.

You want to get a single value from a big table of numbers.

You need to change some values in a list without touching the rest.

You want to analyze only a part of your data, like one column or row.

You want to combine or compare specific parts of two datasets.

You want to filter data based on conditions, like all numbers bigger than 10.

Syntax

NumPy

array[index]
array[start:stop]
array[row_index, column_index]

Indexing starts at 0, so the first item is at position 0.

Slicing uses start (included) and stop (excluded) positions.

Examples

Gets the third item (index 2) from the array, which is 30.

NumPy

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

Gets items from index 1 up to but not including 3, so 20 and 30.

NumPy

arr = np.array([10, 20, 30, 40])
print(arr[1:3])

Gets the item in second row, first column, which is 3.

NumPy

matrix = np.array([[1, 2], [3, 4]])
print(matrix[1, 0])

Sample Program

This program shows how to get and change parts of a 2D array using indexing. It prints specific values and the updated array.

NumPy

import numpy as np

# Create a 2D array
data = np.array([[5, 10, 15],
                 [20, 25, 30],
                 [35, 40, 45]])

# Get the value in the first row, second column
value = data[0, 1]
print(f"Value at row 0, column 1: {value}")

# Get the entire second row
second_row = data[1, :]
print(f"Second row: {second_row}")

# Get the last column
last_column = data[:, 2]
print(f"Last column: {last_column}")

# Change the value at row 2, column 0
data[2, 0] = 100
print("Updated data:")
print(data)

Important Notes

Remember that indexing starts at 0, not 1.

Using slicing (start:stop) does not include the stop index.

Changing values by indexing updates the original array.

Summary

Indexing lets you pick specific data quickly.

You can get single values, rows, columns, or slices.

Indexing helps you change data easily and efficiently.

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