Matplotlib动画直方图

时间:2016-01-30 23:49:19

标签: python animation matplotlib

我正在尝试从下面的代码创建一个动画直方图。我可以为每次创建单独的直方图,但是我无法使用matplotlib.animation函数或通过模拟matplotlib tutorial中的代码来设置结果。

import numpy as np
import matplotlib.pyplot as plt


betas = [] # some very long list
entropy = [] # some very long list

for time in [0.0, 0.5, 1.0, 1.5,  2.0, 2.5, 3.0 , 3.5,  4.0, 4.5  5.0, 5.5,   6.0, 6.5 , 7.0, 7.5,  8.0 , 8,5 , 9.0, 9.5 , 10.0]:

    plt.figure('entropy distribution at time %s ' % time)        
    indexbetas = {i for i, j in enumerate(betas) if j == time}
    desiredentropies = [x for i, x in enumerate(entropy) if i in indexbetas] #the desiredentropies list depends on time

    n, bins, patches = plt.hist(desiredentropies, 20, alpha=0.75 , label = 'desired entropies')   

plt.xlabel(r"$S_{(\time=%d)}$" % time, fontsize=20)
plt.ylabel('Frequency of entropies')


plt.legend()
plt.grid(True)
plt.show()

我特别努力地提供我的desiredentropies列表,这取决于动画的time列表中的元素。

1 个答案:

答案 0 :(得分:6)

试试这个。这基本上只是利用FuncAnimation来更新直方图。查看animation documentation以了解有关该功能的各种参数的更多信息,以控制更新速度和类似内容。

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

n = 100
number_of_frames = 10
data = np.random.rand(n, number_of_frames)

def update_hist(num, data):
    plt.cla()
    plt.hist(data[num])

fig = plt.figure()
hist = plt.hist(data[0])

animation = animation.FuncAnimation(fig, update_hist, number_of_frames, fargs=(data, ) )
plt.show()

我们在这里做的是调用一个函数update_hist来处理更新直方图并在每一步显示新数据。我们通过清除轴然后使用提供的num(即当前帧编号)索引到我们的数据来完成此操作。