Pyplot:带滑块的imshow 3D阵列

时间:2019-03-28 15:21:16

标签: python matplotlib

我有一个3d数组

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

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

并且我想可视化A

中的每个5x5图像
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阵列配合使用?

1 个答案:

答案 0 :(得分:0)

事实证明,问题根本不在于滑块,我只需要将滑块返回的值转换为int

替换为

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

做到了