Как построить 2d математических векторов с matplotlib?

Как мы можем построить 2D математические векторы с matplotlib? Кто-нибудь есть пример или предложение об этом?

У меня есть пара векторов, хранящихся в виде массивов 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]])

Edit:

Я просто добавил график окончательного ответа tcaswell для всех, кто интересуется выходом и хочет построить 2d векторы с matplotlib: enter image description here

Ответы

Ответ 1

Предложение в комментариях по 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()

Ответ 2

Это довольно просто. Надеюсь, этот пример поможет.

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, как если бы это был обычный кортеж для большинства примеров.