问python用表面图和第四个变量的滑块可视化4d数据

EN

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


nx = 100
ny = 100
nz = 100

data = np.random.rand(nx,ny,nz)

fig = plt.figure(1, figsize=(6,6))
main_ax = fig.add_axes([0.1,0.2,0.8,0.7])
slider_ax  = fig.add_axes([0.1,0.1,0.8,0.05])

main_ax.imshow(data[:,:,0], aspect='auto')


my_slider = Slider(slider_ax, 'layer', 0, nz, valinit=0, valfmt='%d')

def update(val):
    main_ax.imshow(data[:,:,int(val)], aspect='auto')
    plt.draw()

my_slider.on_changed(update)
plt.show()