Python：矩阵的形状和 imshow()

Question

我有一个 3-D 数组 ar。

print shape(ar)  # --> (81, 81, 256) 

我想绘制这个数组。

fig = plt.figure()
ax1 = fig.add_subplot(111)
for i in arange(256):
    im1 = ax1.imshow(ar[:][:][i])
    plt.draw()
    print i

我收到此错误消息：

    im1 = ax1.imshow(ar[:][:][i])
IndexError: list index out of range

为什么我收到此奇怪的消息？该图的尺寸为 81 x 256，与预期的 81 x 81 不同。但为什么呢？

Answer 1

执行：

ar[:,:,i]

语法 ar[:] 制作 ar 的副本（切片其所有元素），因此 ar[:][:][i] 在语义上等同于 ar[i]。这是一个 81*256 矩阵，因为 ndarray 是嵌套列表。