數(shù)組索引等同于訪問數(shù)組元素。
您可以通過引用其索引號(hào)來訪問數(shù)組元素。
NumPy 數(shù)組中的索引以 0 開頭,這意味著第一個(gè)元素的索引為 0,第二個(gè)元素的索引為 1,以此類推。
下面我們先來看看一維數(shù)組的索引操作:
>>> import numpy as np >>> arr = np.array([1, 2, 3, 4]) >>> print(arr[0]) # 打印數(shù)組第一個(gè)元素 1 >>> print(arr[1]) # 打印數(shù)組第二個(gè)元素 2 >>> print(arr[2] + arr[3]) # 數(shù)組第三和第四個(gè)元素相加 7
要訪問二維數(shù)組中的元素,我們可以使用逗號(hào)分隔的整數(shù)表示元素的維數(shù)和索引。
下面我們先來看看二維數(shù)組的索引操作:
>>> import numpy as np >>> arr = np.array([[1,2,3,4,5], [6,7,8,9,10]]) >>> print('2nd element on 1st dim: ', arr[0, 1]) # 訪問第一維中的第二個(gè)元素 2nd element on 1st dim: 2 >>> print('5th element on 2nd dim: ', arr[1, 4]) # 訪問第二維中的第五個(gè)元素: 5th element on 2nd dim: 10
要訪問 3-D 數(shù)組中的元素,我們可以使用逗號(hào)分隔的整數(shù)來表示元素的維數(shù)和索引。
>>> import numpy as np >>> arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]) >>> print(arr[0, 1, 2]) 6
第一個(gè)數(shù)字代表第一個(gè)維度,其中包含兩個(gè)數(shù)組:
[[1, 2, 3], [4, 5, 6]] 和 [[7, 8, 9], [10, 11, 12]], 由于我們選擇了 0,所以剩下第一個(gè)數(shù)組:[[1, 2, 3], [4, 5, 6]]。
第二個(gè)數(shù)字代表第二維,它也包含兩個(gè)數(shù)組:[1, 2, 3]和 [4, 5, 6]因?yàn)槲覀冞x擇了 1,所以剩下第二個(gè)數(shù)組:[4, 5, 6]。
第三個(gè)數(shù)字代表第三維,其中包含三個(gè)值:4、5、6
由于我們選擇了 2,因此最終得到第三個(gè)值:6
使用負(fù)索引從尾開始訪問數(shù)組。
打印第二個(gè)維中的的最后一個(gè)元素:
>>> import numpy as np >>> arr = np.array([[1,2,3,4,5], [6,7,8,9,10]]) >>> print('第二維度的最后一個(gè)元素: ', arr[1, -1]) 第二維度的最后一個(gè)元素: 10