时间戳

首页 > 解决方案 > Pyplot：带有滑块的 imshow 3D 数组

python - Pyplot：带有滑块的 imshow 3D 数组

问题描述

我有一个 3d 数组

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 阵列一起使用？

标签： pythonmatplotlib

解决方案

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

替换为

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

成功了

推荐阅读