Использование plt.quiver
Как использовать стрелку или колчан matplotlib со списком, подобным следующему:
#X, Y, U, V
data = np.array([[1, 2, 0.5, 1],
[2, 3, 1, 1.5],
[3, 4, 1.5, 1.75],
[4, 5, 2, 2],
[5, 6, 2.5, 2.5],
[6, 8, 3, 3],
[7, 9, 3.5, 3.5],
[8, 20, 4, 4],
[9, 2, 4, 4.5]])
X = data[:, 0]
Y = data[:, 1]
U = data[:, 2]
V = data[:, 3]
Цель состоит в том, чтобы нарисовать стрелки от (X,Y) до (U,V) в одном кадре с Xlim, Ylim = (30,30).
Примеры, которые я вижу в интернете, используют сетку для колчана. Есть намеки?
1 ответ
Решение
В найденных вами примерах сетка используется не как сетка, а для поэлементного построения графика. Независимо от того, какую форму имеют массивы, элементы в одинаковых позициях принадлежат друг другу.
import matplotlib.pyplot as plt
import numpy as np
plt.figure()
plt.title('Arrows scale with plot width, not view')
#X, Y, U, V
data = np.array([[1, 2, 0.5, 1],
[2, 3, 1, 1.5],
[3, 4, 1.5, 1.75],
[4, 5, 2, 2],
[5, 6, 2.5, 2.5],
[6, 8, 3, 3],
[7, 9, 3.5, 3.5],
[8, 20, 4, 4],
[9, 2, 4, 4.5]])
X = data[:, 0]
Y = data[:, 1]
# If you want to plot vectors starting at X,Y with components U,V
U = data[:, 2]
V = data[:, 3]
# If you want to plot vectors starting at X,Y and ending at U,V
U = data[:, 2] - X
V = data[:, 3] - Y
Q = plt.quiver(X, Y, U, V, units='width')
plt.xlim(0, 30)
plt.ylim(0, 30)
plt.show()
В следующий раз, пожалуйста, предоставьте ваш код в удобном формате.