在matplotlib中使用动画的Colormap问题

时间:2013-05-20 19:56:35

标签: python animation matplotlib color-mapping h5py

我使用matplotlib.animation为名为arr的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()

1 个答案:

答案 0 :(得分:1)

我认为您要将set_clim()替换为

p.autoscale()

没有参数,set_clim()是无操作。

也就是说,在动画中间改变你的色阶似乎很容易让人误解。

您还应该使用set_data代替set_array(根据文档)。