如何用Matplotlib Animation实现箭头固定位置闪烁效果？

我是Python新手，刚掌握Matplotlib动画的基本原理，但无法实现箭头固定指向红色圆圈的闪烁效果（两个圆圈保持静止，不参与动画）。我曾尝试复制箭头坐标设置2帧，但未成功；参考Stack Overflow的示例后，ax.clear不符合需求，ax.patches.remove执行时出现‘IndexError: list index out of range’错误，箭头仍保持静止，恳请相关建议。

原示例代码：

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

fig = plt.figure()
ax = fig.gca()

# Axes labels and title are established
ax = fig.gca()
ax.set_xlabel('x')
ax.set_ylabel('y')

ax.set_ylim(-20, 20)
ax.set_xlim(-20, 20)

# Drawing static circles:
circle1 = plt.Circle((10, 10), 3, color='red')
circle2 = plt.Circle((-10, -4), 2, color='blue')
ax.add_patch(circle1)
ax.add_patch(circle2)

# Coordinates of the arrow:
x = np.array([2, 2])
y = np.array([2, 2])
dx = x*2
dy = y*2

patch = patches.Arrow(x[0], y[0], dx[0], dy[0])


def init():
    ax.add_patch(patch)
    return patch,


def animate(i):
    global patch
    ax.patches.remove(patch)
    patch = patches.Arrow(x[i], y[i], dx[i], dy[i])
    ax.add_patch(patch)
    return patch,


anim = FuncAnimation(fig, animate, init_func=init,
                     frames=len(x), interval=200)

anim.save('222c.gif', writer='pillow')
plt.show()
plt.close()

问题分析 #

1. 箭头无变化：原代码中x、y数组值完全相同，计算出的dx、dy也一致，导致两帧箭头位置无差异，自然看不到动画效果。
2. 索引错误：ax.patches.remove(patch)会在第二次执行时报错，因为第一次移除后，patch已不在ax.patches列表中；且列表中包含静态圆圈，直接操作易误删元素。
3. ax.clear不符合需求：该方法会清除轴上所有元素，包括需要保留的静态圆圈。

解决方案：切换箭头可见性实现闪烁 #

通过控制箭头的可见性来实现闪烁，无需频繁删除/添加patch，同时确保箭头始终指向红圈：

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

fig = plt.figure()
ax = fig.gca()

# 设置坐标轴
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_ylim(-20, 20)
ax.set_xlim(-20, 20)

# 绘制静态圆圈
circle1 = plt.Circle((10, 10), 3, color='red')
circle2 = plt.Circle((-10, -4), 2, color='blue')
ax.add_patch(circle1)
ax.add_patch(circle2)

# 固定箭头起点，计算指向红圈的向量
arrow_start = (2, 2)
red_circle_center = (10, 10)
dx = red_circle_center[0] - arrow_start[0]
dy = red_circle_center[1] - arrow_start[1]

# 创建箭头patch并添加到轴上
arrow_patch = patches.Arrow(*arrow_start, dx, dy, color='black')
ax.add_patch(arrow_patch)

def animate(i):
    # 每隔一帧切换箭头可见性，实现闪烁
    arrow_patch.set_visible(i % 2 == 0)
    return arrow_patch,

# 生成循环动画，frames设为100可循环100次，设为None则无限循环
anim = FuncAnimation(fig, animate, frames=100, interval=500, blit=True)

anim.save('arrow_blink.gif', writer='pillow')
plt.show()
plt.close()

关键改动说明 #

• 固定指向红圈：通过红圈中心坐标减去箭头起点坐标，计算出箭头的偏移量dx、dy，确保箭头始终指向红圈。
• 闪烁逻辑：利用i % 2 == 0判断帧序号，切换箭头的可见状态，避免频繁操作ax.patches列表，杜绝索引错误。
• 性能优化：设置blit=True只更新变化的元素，提升动画流畅度。

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