Использование 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()

В следующий раз, пожалуйста, предоставьте ваш код в удобном формате.

Другие вопросы по тегам