在 matplotlib 中使用动画的颜色图问题

Question

我使用matplotlib.animation 为名为@9​​87654322@ 的3D 数组中的数据设置动画。我使用 h5py 库从 h5 文件中读取数据，一切正常。但是在使用动画时，颜色图卡在数据范围的第一帧，经过一些步骤后，它在绘图时显示未标准化的颜色。

这是我的代码：

import numpy as np
import h5py
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import matplotlib.cm as cm

f = h5py.File('ez.h5','r')
arr = f["ez"][:,:,:]
f.close()

fig = plt.figure()

i = 0
p = plt.imshow(arr[:,:,0], interpolation='bilinear', cmap=cm.RdYlGn)

def updatefig(*args):
    global i
    i += 1
    if (i==333):
        i = 0
    p.set_array(arr[:,:,i])
    plt.clim()
    return p,

ani = animation.FuncAnimation(fig, updatefig, interval=50, blit=True)
plt.show()

Answer 1

我想你想用

替换set_clim()

p.autoscale()

没有参数，set_clim() 是一个空操作。

也就是说，在动画中间更改色阶似乎非常具有误导性。

您还应该使用set_data 而不是set_array（根据文档）。