如何为 Seaborn 热图或相关矩阵（Matplotlib）创建动画？

要为 Seaborn 热图或相关矩阵创建动画，我们可以采取以下步骤 −

  • 设置图片大小并调整子图之间以及周围的内边距。
  • 创建一个新图片或激活一个现有的图片。
  • 制作一个维度元组。
  • 制作 Seaborn 热图。
  • 为第一个热图创建一个 init() 方法。
  • 使用 FuncAnimation() 类创建一个动画，并通过反复调用函数 animate 来创建随机数据集并创建一个热图。
  • 要显示图片，使用 show() 方法。

示例

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
from matplotlib import animation

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

fig = plt.figure()
dimension = (5, 5)
data = np.random.rand(dimension[0], dimension[1])
sns.heatmap(data, vmax=.8)

def init():
    sns.heatmap(np.zeros(dimension), vmax=.8, cbar=False)

def animate(i):
    data = np.random.rand(dimension[0], dimension[1])
    sns.heatmap(data, vmax=.8, cbar=False)

anim = animation.FuncAnimation(fig, animate, init_func=init, frames=20, repeat=False)

plt.show()

输出

更新于： 04-Aug-2021

2K+ 浏览

开启您的职业生涯

完成课程以获得认证

开始
广告