Mastering NumPy Indexing and Slicing Techniques for Efficient Data Manipulation | tutorials 5 | Better4Code
What is NumPy Indexing and Slicing? NumPy is a powerful library for scientific computing in Python. It provides support for arrays, which are the fundamental data structure for numerical computing. NumPy arrays are similar to Python lists, but they are more efficient and optimized for numerical operations. In this article, we’ll explore how to index and slice NumPy arrays.
Indexing a One-Dimensional Array
To access a single element of a one-dimensional NumPy array, we use an integer index enclosed in square brackets. For example, consider the following array:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
To access the first element of the array, we use the index 0:
print(arr[0])
The output will be:
1
We can also use negative indexing to access elements from the end of the array. For example, to access the last element of the array, we use the index -1:
print(arr[-1])
The output will be:
5
Slicing a One-Dimensional Array
We can also use slicing to access a range of elements in a one-dimensional NumPy array. Slicing uses the colon (:) operator to specify the start, stop, and step of the slice. For example, to get the first three elements of the array, we use the following slice:
print(arr[0:3])
The output will be:
[1 2 3]
Note that the end index is not included in the slice. We can also use negative indices in the slice:
print(arr[-3:-1])
The output will be:
[3 4]
Indexing a Multi-Dimensional Array
In a multi-dimensional NumPy array, we need to specify the indices for each dimension separated by commas. For example, consider the following two-dimensional array:
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
To access the element in the second row and third column, we use the indices (1, 2):
print(arr[1, 2])
The output will be:
6
We can also use slices to access a range of elements in a multi-dimensional array. For example, to get the first two rows and all the columns, we use the following slice:
print(arr[0:2, :])
The output will be:
[[1 2 3]
[4 5 6]]
To get all the rows and the first two columns, we use the following slice:
print(arr[:, 0:2])
The output will be:
[[1 2]
[4 5]
[7 8]]