python - 如何显示用 quiver 创建的箭头图例？

Question

如何放置使用pyplot.quiver或pyplt.annotate方法绘制的箭头图例？

我目前正在使用以下代码：

import matplotlib.pyplot as plt
from numpy.random import *


x, y  = [1, 2, 3], [0.5, 0.5, 0.5]
u1,v1 = randn(3), randn(3)
u2,v2 = randn(3), randn(3)
u3,v3 = randn(3), randn(3)
QV1 = plt.quiver(x, y, u1, v1, color='r')
QV2 = plt.quiver(x, y, u2, v2, color='b')
QV3 = plt.quiver(x, y, u3, v3, color='g')

我想放一个简单的图例，它以一种简单的方式显示三种不同颜色的箭头，

plt.plot(x, y, color='r', label='red')
...
plt.legend()

Answer 1

您可以使用quiverkey功能为箭头制作图例。

plt.quiverkey(QV1, 1.2, 0.515, 2, 'arrow 1', coordinates='data')
plt.quiverkey(QV2, 1.2, 0.520, 2, 'arrow 2', coordinates='data')
plt.quiverkey(QV3, 1.2, 0.525, 2, 'arrow 3', coordinates='data')