Matplotlib 3D Quiver 图使线条颜色正确，但箭头颜色错误

Question

我正在尝试绘制 x、y 和 z 方向上三个箭头的颤动图，箭头颜色为绿色、红色和蓝色。由于某种原因，线条的颜色是正确的，但箭头的颜色是错误的，我不知道如何修复。这是我的代码：

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

fig = plt.figure()
ax = fig.gca(projection='3d')
cols = ['r', 'g', 'b']

quivers = ax.quiver([0,0,0],[0,0,0],[0,0,0],[1,0,0],[0,1,0],[0,0,1], colors=cols)

ax.set_xlim3d([-2.0, 2.0])
ax.set_xlabel('X')

ax.set_ylim3d([-2.0, 2.0])
ax.set_ylabel('Y')

ax.set_zlim3d([-2, 2])
ax.set_zlabel('Z')
plt.show()

Answer 1

在 中quiver，如果将颜色列表指定为 option colors=..，则颜色循环对所有线段都有效，包括箭头的所有部分。

要获得 3 个不同颜色的箭头，您可以使用 3 个quiver语句，每个语句具有不同的颜色。

方法一

ax.quiver([0],[0],[0],[1],[0],[0], colors='r')
ax.quiver([0],[0],[0],[0],[1],[0], colors='g')
ax.quiver([0],[0],[0],[0],[0],[1], colors='b')

方法2

fr = [(0,0,0), (0,0,0), (0,0,0)]
to = [(1,0,0), (0,1,0), (0,0,1)]
cr = ["red", "green", "blue"]
for (from_, to_, colr_) in zip(fr,to,cr):
    ax.quiver(*from_, *to_, colors=colr_)