python - 如何用 matplotlib 绘制二维数学向量？

Question

我们如何用 matplotlib 绘制二维数学向量？有人对此有示例或建议吗？

我有几个向量存储为二维 numpy 数组，我想将它们绘制为有向边。

要绘制的向量构造如下:

import numpy as np
# a list contains 3 vectors;
# each list is constructed as the tail and the head of the vector
a = np.array([[0, 0, 3, 2], [0, 0, 1, 1], [0, 0, 9, 9]]) 

---

编辑:

我刚刚添加了 tcaswell 的最终答案图，供任何对输出感兴趣并希望使用 matplotlib 绘制 2d 向量的人:

Answer 1

halex 评论中的建议是正确的，您想使用 quiver (doc)，但您需要稍微调整一下属性。

import numpy as np
import matplotlib.pyplot as plt

soa = np.array([[0, 0, 3, 2], [0, 0, 1, 1], [0, 0, 9, 9]])
X, Y, U, V = zip(*soa)
plt.figure()
ax = plt.gca()
ax.quiver(X, Y, U, V, angles='xy', scale_units='xy', scale=1)
ax.set_xlim([-1, 10])
ax.set_ylim([-1, 10])
plt.draw()
plt.show()