通过索引从Numpy数组中选择一个元素或子数组

NumPy数组的元素的索引和普通数组一样。第一个元素的索引是0，最后一个元素的索引是n-1，其中n是元素的总数。

从NumPy数组中选择一个元素

这些ndarrays的每个元素都可以使用其索引号进行访问。

例子：下面的代码显示了如何访问NumPy数组的一个元素。

import numpy as np
  
# NumPy Array
numpyArr = np.array([1, 2, 3, 4])
print("numpyArr[0] =", numpyArr[0])
print("numpyArr[-1] =", numpyArr[-1])

输出:

numpyArr[0] = 1
numpyArr[-1] = 4

在第一种情况下，我们使用其索引号访问数组的第一个元素。在第二种情况下，我们使用负数的索引来访问数组的最后一个元素。

从NumPy数组中选择一个*子数组（切分）

为了得到一个子数组，我们传递一个片断来代替元素索引。

语法:

numpyArr[x:y]

这里的x和y是所需子数组的起始和最后索引。

示例:

import numpy as np
  
# NumPy Array
numpyArr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
  
# the slice 3:6 is passed instead 
# of index
print("Sub-Array=", numpyArr[3:6])

输出:

Sub-Array= [4 5 6]

一个从第3个索引开始到第6个索引的子数组被选中（不包括最后的第6个索引）。你可以将起始索引留空，从第一个元素开始切入一个子数组。

例子：下面的代码从第一个元素开始选择一个子数组。

import numpy as np
  
numpyArr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
  
# works same as 0:6
print("Sub-Array=", numpyArr[:6])

输出:

Sub-Array= [1 2 3 4 5 6]

同样的，把冒号左边留空会得到一个直到最后一个元素的数组。

例子：下面的代码选择一个子数组，从一个特定的索引开始，直到最后一个索引。

import numpy as np
  
numpyArr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
  
# same as 3:9 or 3:n, where n is
# the length of array
print("Sub-Array=", numpyArr[3:])