使用matplotlib.animation.FuncAnimation实现FancyArrows动画时Python崩溃求助

问题根源 #

原代码运行后崩溃的核心原因是每次动画帧都重复创建新的箭头Patch对象，且未正确复用已有资源，导致内存持续累积，最终触发Python崩溃。此外代码存在集合管理混乱、重复计算等冗余问题。

优化后的代码 #

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

# 相量参数定义
Iamp = 10
f = 50
N = 7
# 预生成所有相位偏移
PHIs = 2 * np.pi * np.arange(N) / N
# 预计算所有时间点的相量数据（避免实时计算）
t = np.linspace(0, 1/f, 100)
phases = 2 * np.pi * f * t[:, np.newaxis] + PHIs
dx = Iamp * np.cos(phases)
dy = Iamp * np.sin(phases)

# 初始化绘图
fig, ax = plt.subplots()
ax.set_ylim(-Iamp, Iamp)
ax.set_xlim(-Iamp, Iamp)
ax.set_aspect('equal', adjustable='box')

# 预创建所有箭头对象，复用而非每次重建
arrows = []
for phi in PHIs:
    # 初始位置
    init_dx = Iamp * np.cos(phi)
    init_dy = Iamp * np.sin(phi)
    arrow = patches.FancyArrow(0, 0, init_dx, init_dy,
                               head_width=Iamp/3, head_length=Iamp/10,
                               length_includes_head=True, animated=True)
    ax.add_patch(arrow)
    arrows.append(arrow)

def animate(frame_idx):
    # 更新每个箭头的终点坐标，而非重建对象
    for arrow, x, y in zip(arrows, dx[frame_idx], dy[frame_idx]):
        arrow.set_data(x=0, y=0, dx=x, dy=y)
    return arrows

# 创建动画，设置blit=True提升渲染效率
anim = animation.FuncAnimation(fig, animate,
                               frames=len(t), interval=150,
                               blit=True)

plt.show()

关键优化说明 #

• 复用对象避免内存泄漏：提前创建所有箭头Patch对象，动画帧仅更新坐标，不再重复创建新对象，彻底解决内存累积导致的崩溃问题。
• 预计算数据提升性能：一次性计算所有时间点的相量坐标，避免动画实时计算三角函数，降低CPU负载。
• 正确使用动画属性：设置animated=True并开启blit=True，让Matplotlib仅渲染变化的部分，大幅提升动画流畅度。
• 简化集合管理：直接管理单个箭头对象，避免PatchCollection的混乱使用，代码逻辑更清晰。
• 参数规范化：用Iamp的比例设置箭头头部尺寸，避免硬编码数值，适配不同幅值的相量。

内容的提问来源于stack exchange，提问作者Victor Rodal