杨桃的Python进阶讲座17——数组array（七）三维数组和n维数组的索引和取值（配详细图解）

本人CSDN博客专栏：https://blog.csdn.net/yty_7

Github地址：https://github.com/yot777/

 

三维数组的索引和取值

创建一个numpy三维数组z，如下所示：

>>> import numpy as np
>>> z=np.array([[[1,2,3,4],[5,6,7,8]],[[9,10,11,12],[13,14,15,16]]])
>>> print(z)
[[[ 1  2  3  4]
  [ 5  6  7  8]]

 [[ 9 10 11 12]
  [13 14 15 16]]]

三维数组没有了矩阵的行、列概念，但我们可以用shape()函数查看它的维度信息：

>>> np.shape(z)
(2, 2, 4)

三维数组z的维度是(2, 2, 4)，是不是挺难理解的呢！

其实把(2, 2, 4)变型为(2, (2, 4))，即2个(2, 4)矩阵，就成了三维数组啦！

图示如下，可理解为三维数组z是由2个(2, 4)的矩阵平面构成的

我们继续使用二维数组的索引方式来查看三维数组，如下：

如果[ ]以数字或者单冒号开头，表示取矩阵平面的全部元素，举例：

>>> z[0]             #取第0个矩阵平面的元素
array([[1, 2, 3, 4],
       [5, 6, 7, 8]])
>>> z[1]             #取第1个矩阵平面的元素
array([[ 9, 10, 11, 12],
       [13, 14, 15, 16]])
>>> z[:1]            #取从0开始到1-1=0个矩阵平面的元素
array([[[1, 2, 3, 4],
        [5, 6, 7, 8]]])

图示：

如果[ ]以<