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 х 256 и не соответствует ожидаемому 81 х 81. Но почему?

Answer 1

Do:

ar[:,:,i]

Синтаксис ar[:] делает копию ar (разрезает все его элементы), поэтому ar[:][:][i] семантически эквивалентно ar[i].Это матрица 81 * 256, так как ndarrays - это вложенные списки.