如何更新3D quiver plot箭头并解决动画旧箭头重叠问题？

Matplotlib的3D quiver plot返回的是Line3DCollection对象，不像2D quiver那样提供set_UVC()方法直接更新箭头数据。如果每帧重新绘制quiver却不清理旧的集合，就会导致箭头重叠堆积。以下是两种可行的解决方案：

方法1：每帧移除旧箭头集合后重新绘制 #

这种方法简单直观，适合中小规模数据集：

• 保存初始绘制的Line3DCollection对象
• 在动画更新函数中，先调用ax.collections.remove()移除旧箭头集合
• 基于新数据重新绘制quiver，并更新保存的集合对象

代码示例：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.animation as animation
import numpy as np

# 设置画布和3D轴
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.set_xlim(-2, 2)
ax.set_ylim(-2, 2)
ax.set_zlim(-2, 2)

# 初始化网格数据
x, y, z = np.meshgrid(np.arange(-2, 2, 1), np.arange(-2, 2, 1), np.arange(-2, 2, 1))
u = np.zeros_like(x)
v = np.zeros_like(y)
w = np.zeros_like(z)

# 初始绘制3D quiver，保存返回的集合对象
quiver_obj = ax.quiver(x, y, z, u, v, w, length=0.5)

def update(frame):
    global quiver_obj
    # 移除旧的箭头集合
    ax.collections.remove(quiver_obj)
    # 更新箭头数据（示例让z方向分量随帧变化）
    w = np.sin(frame * 0.1 + x.flat).reshape(x.shape)
    # 重新绘制quiver并更新对象引用
    quiver_obj = ax.quiver(x, y, z, u, v, w, length=0.5)
    return quiver_obj,

# 创建并显示动画
ani = animation.FuncAnimation(fig, update, frames=20, interval=200, blit=True)
plt.show()

方法2：直接更新Line3DCollection的segments属性 #

这种方法效率更高，无需反复创建和删除对象，适合大规模数据集：

• 利用Matplotlib内部的_get_quiver_segments()方法生成新的箭头线段数据
• 直接替换Line3DCollection对象的segments属性，实现箭头数据更新

代码示例：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.animation as animation
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.set_xlim(-2, 2)
ax.set_ylim(-2, 2)
ax.set_zlim(-2, 2)

x, y, z = np.meshgrid(np.arange(-2, 2, 1), np.arange(-2, 2, 1), np.arange(-2, 2, 1))
u = np.zeros_like(x)
v = np.zeros_like(y)
w = np.zeros_like(z)

# 定义quiver的参数，确保每次生成segments时参数一致
quiver_params = {'length': 0.5, 'color': 'darkorange'}
# 初始绘制quiver
quiver_obj = ax.quiver(x, y, z, u, v, w, **quiver_params)

def update(frame):
    # 更新箭头数据
    w = np.cos(frame * 0.1 + x.flat + y.flat).reshape(x.shape)
    # 生成新的箭头线段数据
    new_segments = ax._get_quiver_segments(x, y, z, u, v, w, **quiver_params)
    # 更新集合的segments属性
    quiver_obj.set_segments(new_segments)
    # 若需更新颜色，可调用set_color()方法
    # quiver_obj.set_color(np.random.rand(len(x.flat), 3))
    return quiver_obj,

ani = animation.FuncAnimation(fig, update, frames=20, interval=200, blit=True)
plt.show()

注意事项 #

• 使用方法1时，若动画包含其他绘图元素，需确保返回所有需要更新的艺术家对象，否则blit=True可能导致显示异常
• 方法2中_get_quiver_segments()是Matplotlib的内部方法，未来版本可能有变动，若出现兼容性问题，可自行构造箭头的线段数据
• 在Jupyter Notebook中运行时，需添加plt.rcParams['animation.html'] = 'jshtml'以正常显示动画

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