numpy 阵列存取
示例
切片的语法i:j:k,其中i的起始索引(含),j是止动指数(不包括)和k是步长。像其他python数据结构一样,第一个元素的索引为0:
x = np.arange(10) x[0] # Out: 0 x[0:4] # Out: array([0, 1, 2, 3]) x[0:4:2] # Out:array([0, 2])
负值从数组末尾开始计数。 -1因此访问数组中的最后一个元素:
x[-1] # Out: 9 x[-1:0:-1] # Out: array([9, 8, 7, 6, 5, 4, 3, 2, 1])
多维数组可以通过指定每个以逗号分隔的维来访问。以前的所有规则均适用。
x = np.arange(16).reshape((4,4)) x # Out: # array([[ 0, 1, 2, 3], # [ 4, 5, 6, 7], # [ 8, 9, 10, 11], # [12, 13, 14, 15]]) x[1,1] # Out: 5 x[0:3,0] # Out: array([0, 4, 8]) x[0:3, 0:3] # Out: # array([[ 0, 1, 2], # [ 4, 5, 6], # [ 8, 9, 10]]) x[0:3:2, 0:3:2] # Out: # array([[ 0, 2], # [ 8, 10]])