我正在尝试从下面的代码创建一个动画直方图。我可以为每次创建单独的直方图,但是我无法使用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
列表中的元素。
答案 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
(即当前帧编号)索引到我们的数据来完成此操作。