如何在Matplotlib的3D绘图中实现箭头沿曲线路径放置？

Matplotlib的FancyArrowPatch本身仅支持2D场景，但可以通过将3D坐标投影到2D画布的方式实现3D箭头效果。下面是具体的解决方案：

实现步骤 #

1. 自定义支持3D的箭头类，继承FancyArrowPatch，通过3D投影转换处理坐标
2. 在参数曲线上选择需要添加箭头的位置，调用自定义类绘制箭头

完整示例代码 #

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.patches import FancyArrowPatch
from mpl_toolkits.mplot3d.proj3d import proj_transform

class Arrow3D(FancyArrowPatch):
    def __init__(self, xs, ys, zs, *args, **kwargs):
        super().__init__((0,0), (0,0), *args, **kwargs)
        self._verts3d = xs, ys, zs

    def draw(self, renderer):
        xs3d, ys3d, zs3d = self._verts3d
        xs, ys, zs = proj_transform(xs3d, ys3d, zs3d, self.axes.M)
        self.set_positions((xs[0], ys[0]), (xs[1], ys[1]))
        super().draw(renderer)

# 原有绘图代码
z = np.linspace(0,10,100)
x = np.sin(z)
y = np.cos(z)

fig = plt.figure()
ax = fig.add_subplot(projection='3d')
ax.plot(x, y, z, label='parametric curve')

# 在曲线上添加箭头：比如在z=2的位置
idx = np.argmin(np.abs(z - 2))
# 箭头起点与终点：取曲线上相邻点确定指向
start = (x[idx], y[idx], z[idx])
end = (x[idx+5], y[idx+5], z[idx+5])

arrow = Arrow3D([start[0], end[0]], [start[1], end[1]], [start[2], end[2]],
                arrowstyle="-|>", color="red", linewidth=2)
ax.add_artist(arrow)

# 可添加多个箭头
idx2 = np.argmin(np.abs(z - 6))
start2 = (x[idx2], y[idx2], z[idx2])
end2 = (x[idx2+5], y[idx2+5], z[idx2+5])
arrow2 = Arrow3D([start2[0], end2[0]], [start2[1], end2[1]], [start2[2], end2[2]],
                 arrowstyle="-|>", color="blue", linewidth=2)
ax.add_artist(arrow2)

ax.legend()
plt.show()

关键说明 #

• Arrow3D类通过proj_transform将3D坐标转换为画布的2D坐标，让FancyArrowPatch能正确渲染在3D空间的对应位置
• 箭头的样式、颜色、粗细等参数可通过arrowstyle、color、linewidth等调整，与FancyArrowPatch参数完全兼容
• 可根据需求选择曲线上任意位置添加箭头，只需定位对应点的3D坐标即可

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