极地Matplotlib图中的箭头

时间:2014-06-01 16:44:19

标签: python matplotlib plot polar-coordinates

我试图绘制一系列R-L-C电路中电阻,电容和电感两端电压的相量。我已经完成了所有的计算,我可以用正常的ax.plot(theta,r,....)获得一个不错的情节。

我想让相量矢量看起来像箭头。我一直在尝试使用ax.arrow(0,0,theta,magnitude),但它看起来像一条线。我写的代码的要点是:GIST

我创建的图片是

我试着按照我在this list找到的例子,因为它与我想要完成的事情非常相似,它产生了以下图像:

当我在计算机上运行代码时,我得到了

我在Xubuntu 14.04上运行matplotlib 1.3.1。我确实看到我使用的示例是在2009年使用matplotlib 0.99。

非常感谢任何帮助。

1 个答案:

答案 0 :(得分:4)

箭头尺寸太大了,这个:

import matplotlib
import numpy as np
import matplotlib.pyplot as plt

print "matplotlib.__version__   = ", matplotlib.__version__
print "matplotlib.get_backend() = ", matplotlib.get_backend()


# radar green, solid grid lines
plt.rc('grid', color='#316931', linewidth=1, linestyle='-')
plt.rc('xtick', labelsize=15)
plt.rc('ytick', labelsize=15)

# force square figure and square axes looks better for polar, IMO
width, height = matplotlib.rcParams['figure.figsize']
size = min(width, height)
# make a square figure
fig = plt.figure(figsize=(size, size))
ax = fig.add_axes([0.1, 0.1, 0.8, 0.8], polar=True, axisbg='#d5de9c')

r = np.arange(0, 3.0, 0.01)
theta = 2*np.pi*r
ax.plot(theta, r, color='#ee8d18', lw=3)
ax.set_rmax(2.0)
plt.grid(True)

ax.set_title("And there was much rejoicing!", fontsize=20)
#This is the line I added:
arr1 = plt.arrow(0, 0.5, 0, 1, alpha = 0.5, width = 0.015,
                 edgecolor = 'black', facecolor = 'green', lw = 2, zorder = 5)

# arrow at 45 degree
arr2 = plt.arrow(45/180.*np.pi, 0.5, 0, 1, alpha = 0.5, width = 0.015,
                 edgecolor = 'black', facecolor = 'green', lw = 2, zorder = 5)

plt.show()

产地:

enter image description here

更好? :)