iframe-proxy | Sunbelt Computer Software

39 lines (34 loc) · 1.21 KB

def animation_test():

import matplotlib.pyplot as plt

import matplotlib.animation as animation

num_transixal= 4

num_axis= 9

num= num_transixal* num_axis

N = 2

ratio_data: np.ndarray # [72, 9, 8]

fig, ax = plt.subplots()

img = ratio_data[0,]

# plt.imshow(img)

vmax = 100

vmin = 0

h = ax.imshow(img, cmap=plt.get_cmap('CMRmap_r'), origin='lower', interpolation='none', vmin=vmin, vmax=vmax,

animated=True)

ax.set_xticks(range(N * num_transixal))

ax.set_xticklabels(range(1, N * num_transixal+ 1))

ax.set_yticks(range(num_axis))

ax.set_yticklabels(range(1, num_axis+ 1))

ax.set_title(f'frame: 1')

fig.tight_layout()

def update(frame):

img = ratio_data[frame,]

# h.set_data(img)

h.set_array(img)

ax.set_title(f'frame: {frame}')

return h,

# create animation

interval = 100

ani = animation.FuncAnimation(fig, update, frames=range(72), interval=interval, blit=True)

# ani = animation.FuncAnimation(fig, update, frames=frame_iter, interval=interval, blit=False, cache_frame_data=False)

ani.save('temp.gif', writer='pillow', fps=2, dpi=300)

if __name__ == '__main__':

animation_test()

Sunbelt Computer Software