Python剧情地为一系列热图制作动画

Question

我有这个代码遍历神经网络层权重（为每个时期保存），并使用plotly我创建了很好的热图（X - 神经元在层l，Y - 与神经元相关的权重）热图每一层，时代。

我最终想要的是动画（播放与特定网络层相关的热图序列）以观察神经元连接的进展。

有没有人知道如何基于我的代码生成的这些热图集来实现绘图动画（结果应该是mp4或mpeg文件或嵌入绘图动画的html文件）

 for i, epoch in enumerate(files):
    pkl_file = open(os.path.join(os.getcwd(), "weights_visual_raw_data", epoch), 'rb')
    weights = pickle.load(pkl_file)
    pkl_file.close()
    weights = [x for x in weights if x]
    for l, layer in enumerate(weights):
        if l >= len(weights)-2:
            layer_weights = layer[0]
        else:
            layer_weights = layer[1]
        trace = go.Heatmap(z=layer_weights)
        data = [trace]
        plot(data,auto_open=False, filename=os.path.join(os.getcwd(),
                                         "weights_dynamic_visualization",
                                         "layer"+str(l),
                                         "weights_epoch_" + str(i)))