matplotlib 动画中显示完整子图的方法

在使用 matplotlib 创建动画时，当多个子图并排排列时，可能会出现子图被裁剪的问题。具体表现为：

• 默认情况下：子图被裁剪，只显示部分内容。
• 拉伸查看器窗口：子图显示完整，但需要手动拉伸窗口。
• 保存为 mp4：子图被裁剪，无法正常保存完整内容。
  

解决方案

1. 使用 plt.subplots_adjust() 函数调整子图位置

plt.subplots_adjust() 函数可以调整子图的位置，使其在窗口中显示完整。具体使用方法如下：

plt.subplots_adjust(left=0, bottom=0, right=1, top=1,
                   wspace=0, hspace=0)

其中，left、bottom、right、top 分别表示子图距离窗口左、下、右、上的距离，wspace 和 hspace 分别表示子图之间的水平间距和垂直间距。

2. 使用 plt.tight_layout() 函数自动调整子图位置

plt.tight_layout() 函数可以自动调整子图位置，使其在窗口中显示完整。具体使用方法如下：

plt.tight_layout()

3. 使用动画的 blit=True 参数优化动画性能

在创建动画时，可以使用 blit=True 参数优化动画性能。具体使用方法如下：

anim = animation.FuncAnimation(fig, anim_update, frames=frames,
                               interval=3000, blit=True)

4. 使用 ffmpeg 将动画保存为 mp4 文件

在保存动画为 mp4 文件时，可以使用 ffmpeg 命令将动画保存为 mp4 文件。具体使用方法如下：

ffmpeg -i animation.gif -c:v libx264 -crf 20 -pix_fmt yuv420p animation.mp4

其中，-i animation.gif 表示输入的动画文件，-c:v libx264 表示使用的视频编码器，-crf 20 表示视频质量，-pix_fmt yuv420p 表示视频像素格式。

代码例子

下面是一个使用 matplotlib 创建动画并保存为 mp4 文件的代码示例：

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

# 创建子图
fig, (agent_ax, signal_ax) = plt.subplots(1, 2, sharey=True)

# 设置子图的范围
agent_ax.set_ylim(0, 10)
agent_ax.set_xlim(0, 10)
signal_ax.set_ylim(0, 10)
signal_ax.set_xlim(0, 10)

# 创建子图中的数据
agent_data = np.random.rand(10, 10)
signal_data = np.random.rand(10, 10)

# 将数据显示在子图中
agent_mat = agent_ax.matshow(agent_data, vmin=0, vmax=1)
signal_mat = signal_ax.matshow(signal_data, vmin=0, vmax=1)

# 调整子图位置
plt.subplots_adjust(left=0, bottom=0, right=1, top=1,
                       wspace=0, hspace=0)

# 定义动画更新函数
def anim_update(tick):
    # 更新数据
    agent_data = np.random.rand(10, 10)
    signal_data = np.random.rand(10, 10)

    # 将数据显示在子图中
    agent_mat.set_data(agent_data)
    signal_mat.set_data(signal_data)

    return agent_mat, signal_mat

# 创建动画
anim = animation.FuncAnimation(fig, anim_update, frames=100,
                               interval=3000, blit=True)

# 将动画保存为 mp4 文件
anim.save("animation.mp4", fps=5, extra_args=['-vcodec', 'libx264'])

通过运行上面的代码，可以创建一个动画并将其保存为 mp4 文件。动画中的子图将完整显示，不会被裁剪。

总结

在使用 matplotlib 创建动画时，可以通过调整子图位置、优化动画性能或使用 ffmpeg 将动画保存为 mp4 文件等方法解决子图被裁剪的问题。