2013-10-03 238 views
2

如何將箭頭圖例使用pyplot.quiverpyplt.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() 

回答

4

你可以使用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')