Как создать график колчана в matplotlib (с примерами)

График колчана — это тип графика, на котором отображаются стрелки с компонентами направления U и V в декартовых координатах, заданных X и Y.

Мы можем легко создать график колчана в Matplotlib, используя функцию quiver() , которая использует следующий синтаксис:

колчан (x, y, u, v)

Золото:

• x: координаты x расположения стрелок
• y: координаты y расположения стрелок
• u: компоненты x векторов стрелок.
• v: компоненты y векторов стрелок.

В этом руководстве представлено несколько примеров практического использования этой функции.

Пример 1: рисунок колчана с одной стрелой

Следующий код показывает, как отобразить график колчана с помощью одной стрелки:

 import matplotlib. pyplot as plt

#define plots
fig, ax = plt. subplots ()

#define coordinates and directions
x = 0
y = 0
u = 15
v = 3

#create quiver plot
ax. quiver (x, y, u, v)

#display quiver plot
plt. show ()

Пример 2: раскладка колчана с двумя стрелками

Следующий код показывает, как отобразить график колчана с двумя стрелками:

 import matplotlib. pyplot as plt

#define plots
fig, ax = plt. subplots ()

#define coordinates and directions
x = [0, 0]
y = [0, 0]
u = [0, 1]
v = [-2, 0]
#create quiver plot
ax. quiver (x, y, u, v, scale = 10 )

#display quiver plot
plt. show () 

Обратите внимание, что аргумент масштабирования масштабирует стрелки, делая их длиннее, что делает их более заметными на графике.

Пример 3: график колчана с сетчатой сеткой

Следующий код показывает, как отобразить график колчана с помощью сетчатой сетки:

 import matplotlib. pyplot as plt
import numpy as np

#define plots
fig, ax = plt. subplots ()

#define coordinates and directions
x,y = np. meshgrid (np. arange (-2, 2, .1), np. arange (-2, 2, .1))
z = x*np. exp (-x**2 - y**2)
v, u = np. gradient (z, .1, .1)

#create quiver plot
ax. quiver (x, y, u, v)

#display quiver plot
plt. show () 

Полную документацию функции quiver() вы можете найти здесь .