如何使用matplotlib绘制2d数学向量?

时间:2012-09-04 14:06:55

标签: python math matplotlib visualization

我们如何使用matplotlib绘制2D数学向量?有没有人有这方面的例子或建议?

我有几个向量存储为2D 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向量: enter image description here

2 个答案:

答案 0 :(得分:32)

halex评论中的建议是正确的,你想使用箭头(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()

答案 1 :(得分:0)

这很简单。希望这个例子有所帮助。

import matplotlib.pyplot as plt
import numpy as np
x = np.random.normal(10,5,100)
y = 3 + .5*x + np.random.normal(0,1,100)
myvec = np.array([x,y])
plt.plot(myvec[0,],myvec[1,],'ro')
plt.show()

将产生:

enter image description here

要绘制数组,您可以将它们分割成1D向量并绘制它们。我已经阅读了所有不同选项的matplotlib的完整文档。但是你可以将numpy向量看作是大多数例子中的正常元组。