Numpy Fancy Indexing
In earlier section, we discussed
indexing
(arr[2])
and slicing
(arr[:5])
to fetch a single element and subset of array, respectively.In Fancy Indexing, we pass array of indices instead of single scalar(numbers) to fetch elements at different index points. Remember that the shape of the output depends on the shape of the index arrays rather than the shape of the array being indexed
Let’s go through some examples to understand this concept
import numpy as np
rand = np.random.RandomState(42)
# creating 1d array for demonstration
arr1 = rand.randint(100, size=10)
print(f"1D array:\n{arr1}")
#creating 2d array for demonstration
arr2 = rand.randint(100, size=(3,5))
print(f"\n2D array:\n{arr2}")
1D array:
[51 92 14 71 60 20 82 86 74 74]
2D array:
[[87 99 23 2 21]
[52 1 87 29 37]
[ 1 63 59 20 32]]
For 1D array, let’s suppose we want to access elements at index position of
0
, 4
and -1
# method 1
arr1[0],arr1[4],arr1[-1]
(51, 60, 74)
# method 2
i = np.array([0,4,-1])
arr1[i]
array([51, 60, 74])
For 2D array, we need to provide the index position at both axis levels. Here is an example, where we are giving the index position at both row and column. We wan to get first row, first item; second row, second item; and third row, third item:
row_ind = np.array([0,1,2])
col_ind = np.array([0,1,2])
arr2[row_ind,col_ind]
array([87, 1, 59])
As you can see in above example that the shape of output depends on the shape of index array and not the shape of array being indexed, which is 2D array in this example
We will go a step forward and discuss the cases where we need to fetch multiple items from each row, which in return gives us a 2D array in the output
# first, let's change the shape of row_ind
row_ind = row_ind.reshape(3,1)
print(row_ind)
# slicing the arr2
print(arr2[row_ind, col_ind])
[[0]
[1]
[2]]
[[87 99 23]
[52 1 87]
[ 1 63 59]]
In this section, we will combine simple indexing technique with fancy indexing.
**Simple & Fancy indexing: **Let’s fetch
0
, 3
and -1
indexed items in the column, from the first row 0
of arr2
arr2[0,[0,3,-1]]
array([87, 2, 21])
Fancy Indexing & Slicing: Let’s fetch
0
, 3
and -1
indexed item in the column, from the first 0
and second 1
row of arr2
arr2[:2,[0,3,-1]]
array([[87, 2, 21],
[52, 29, 37]])
Let’s fetch
0
, 3
and -1
indexed item in the column, from every other row of arr2
arr2[::2,[0,3,-1]]
array([[87, 2, 21],
[ 1, 20, 32]])
**Boolean Masking and Slicing: **If you haven’t covered boolean masking then you can just skip this section and come back at it again, later.
Let’s suppose, in every other row, we want to fetch all the column elements whose index position is
True
(1
) in the provided array.mask = np.array([1,0,1,0,1], dtype=bool)
arr2[::2,mask]
array([[87, 23, 21],
[ 1, 59, 32]])
In this section, we will use the fancy indexing techniques to replace the values in the array.
Let suppose, for our 1D array
arr1
, we want to modify the 0
, 4
and -1
values to 0
print(f"arr1 before modification: \n{arr1}")
# defining index position where modification needed
i = np.array([0,4,-1])
# modifying the value
arr1[i] = 0
# printing the results
print(f"arr1 after modification: \n{arr1}")
arr1 before modification:
[51 92 14 71 60 20 82 86 74 74]
arr1 after modification:
[ 0 92 14 71 0 20 82 86 74 0]
In addition to setting the value to single number (
0
in above example), we can also use operations like, +=
, -=
etcUsing the same methods, we can also modify the values in 2D array. Let modify the first and last column values of all rows to
0
print(f"arr2 before modification: \n{arr2}")
# modifiyng values
arr2[:,[0,-1]] = 0
print(f"arr2 after modification: \n{arr2}")
arr2 before modification:
[[87 99 23 2 21]
[52 1 87 29 37]
[ 1 63 59 20 32]]
arr2 after modification:
[[ 0 99 23 2 0]
[ 0 1 87 29 0]
[ 0 63 59 20 0]]
Last modified 6mo ago