python - 如果一个轴包含时间，如何在 matplotlib 中绘制箭头？

Question

这里有点困惑:我想通过在预测时间 t(在 x 轴上)以风速 wind 绘制一系列箭头来绘制风力预报( y 轴)和指向风吹来方向的箭头。箭头的尺寸应该是地 block 高度的给定分数，例如任意确定为地 block 高度的 1/15。由于绘图自动缩放到 wind 列表中的最大风速，因此:

arrowlength = np.max(wind) / 15.0

这应该会产生以绘图 Y 为单位的箭头长度。问题似乎是 X 单位与绘图目的不兼容，因为我正在为其使用日期时间。因此，当我执行以下操作时，matplotlib 的内裤出现了问题(这是如果 x 轴不是日期时间时应该工作的代码):

for x,u,v,y in zip(t, windu, windv, wind):
  xcomp = arrowlength * sin(u/v)
  ycomp = arrowlength * cos(u/v)
  arr = Arrow(x-xcomp/2, y-ycomp/2, x+xcomp/2, y+ycomp/2, edgecolor='black')
  ax.add_patch(arr)
arr.set_facecolor('b')

当然，这会在日期时间 t(循环中分别为 x)传递给 Arrow 函数。

我想很多情况都是一样的，这是一个非常普遍的问题:如何将线和其他注释放置到 X 和 Y 中没有相同单位的坐标网格中？

Answer 1

您可以使用 matplotlib.dates ( http://matplotlib.org/api/dates_api.html ) 通过 date2num() 方法将日期转换为轴坐标。在以下示例中，我创建了一个从 (April 5 2015, 10) 到 (April 7 2015, 20) 的箭头:

import matplotlib.pyplot as plt
import matplotlib.dates as md

import datetime

fig = plt.figure()
ax = fig.add_subplot(111)

t0 = datetime.datetime(2015,4,1)
t = [t0+datetime.timedelta(days=j) for j in xrange(20)]
y = range(20)
ax.plot(t, y)

x0 = md.date2num(datetime.datetime(2015,4,5))
y0 = 10
xw = md.date2num(datetime.datetime(2015,4,7)) - x0
yw = 10
arr = plt.Arrow(x0, y0, xw, yw, edgecolor='black')
ax.add_patch(arr)
arr.set_facecolor('b')
fig.autofmt_xdate()
plt.show()