web-dev-qa-db-ja.com

matplotlibで2D数学ベクトルをプロットする方法は?

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]]) 

編集:

出力に興味があり、matplotlibで2Dベクトルをプロットしたい人のために、tcaswellの最終回答のプロットを追加しました。 enter image description here

16
pacodelumberg

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()
32
tacaswell

とても簡単です。この例がお役に立てば幸いです。

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ベクトルを通常のタプルであるかのように扱うことができます。

0
emhart