通过索引从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:])
输出:
Sub-Array= [4 5 6 7 8 9]