将热图应用于极坐标图?

时间:2019-08-08 21:38:32

标签: python-3.x matplotlib

我正在创建风向图,并使用带有单个条的极坐标图来表示风向。现在,我希望该条通过相应地更改其颜色来显示风速,并在图例中具有颜色渐变(具有匹配的速度值)。可以通过matplotlib中的热图/ imshow图完成此操作,还是需要自己使用函数修改颜色?

# Polar plot: wind rose
theta = np.linspace(0,360,17)
labels = ['N', 'NNO','NO', 'ONO', 'O', 'OZO', 'ZO', 'ZZO', 'Z', 'ZZW', 
'ZW', 'WZW', 'W', 'WNW', 'NW', 'NNW']

ax = plt.subplot(projection='polar')
ax.bar((2*np.pi*wDir['Value'])/360, 1, width=0.2, color='r' ,alpha=1)
# set theta axis
ax.set_theta_zero_location('N')
ax.set_theta_direction(-1)
ax.set_thetagrids(theta, labels=labels)
# set radii axis
ax.set_ylim(0,1)
ax.get_yaxis().set_ticklabels([])
ax.set_title('Wind direction', pad=20)
plt.show()

0 个答案:

没有答案