python的数组下标_《python科学技算》之多维数组的下标存取

多维数组的下标是一个长度上与数组维数相同的元组。

a=np.arange(3*4*5).reshape(3,4,5)

lidx=[[1],[2],[3]]

aidx=np.array(lidx)

a[lidx]#a[tuple(lidx)]; a[([0],[1],[2])]

a[aidx] #a[aidx,:,:]

下标元组的所有元素都是切片和整数，它作为下标得到的原始数组的视图，与原始数组共享数据存储空间。

仅有整数数组作为下标，遵循以下步骤：将所有数组进行广播（找维数最大的数组，长度不足前面补1，再找各个维数上shape的最大值，此为广播后数组的最大值，将shape不足的部分cope已有的数组。），对于b=a[i0,i1,i2]来说，b[0,0,0]=a[i0[0,0,0],i1[0,0,0],i2[0,0,0]]，注意，此时的i0,i1,i2都为广播后的数组。

当存在切片时，如果整数数组间没有下标，结果数组的shape为，将整数数组部分替换为广播之后的shape。如a.shape=(3,4,5)， i0与i1广播后为(2,2,3)，则a[1:3,i0,i3].shape()=(2,2,2,3)

如果下标元组中整数数组不是连续的，则数组的shape为整数数组广播后的shape加上切片元素对应的形状，如a[i0,:,i1].shape()=(2,2,3,4),(2,2,3)+a的第一轴的长度为4。

如果使用布尔数组作为下标对象，或下标对象含有数组，相当于用nonzero()（返回布尔数组中值不为0的元素下标）将布尔数组转换为一组整数数组，然后用整数数组进行下标运算