Pyplot: 3D-массив imshow с ползунком

Question

У меня есть трехмерный массив

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider

A = np.random.rand(10,5,5)

, и я хотел бы визуализировать каждое изображение 5x5 в A

for Ai in A:
    plt.imshow(Ai)
    plt.show()

За исключением того, что вместо 5 цифр, как вПример выше, я хотел бы иметь ползунок для переключения между индексами для первой координаты A.

На данный момент я попробовал следующее:

idx0 = 3
l = plt.imshow(A[idx0])

axidx = plt.axes([0.25, 0.15, 0.65, 0.03])
slidx = Slider(axidx, 'index', 0, 9, valinit=idx0, valfmt='%d')

def update(val):
    idx = slidx.val
    l.set_data(A[idx])
    fig.canvas.draw_idle()
slidx.on_changed(update)

plt.show()

Но когдаЯ использую ползунок, изображение на графике не меняется, и я получаю сообщение

IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

Как мне заставить свой ползунок работать с моим 3d-массивом?

Answer 1

Оказывается, проблема не с ползунком вообще, мне просто нужно было преобразовать значение, возвращаемое ползунком в int.

Замена на

l.set_data(A[int(idx)])

делает трюк