Matplotlib - 箭羽图



箭羽图以箭头显示速度矢量,箭头分量为 (u,v),位置为 (x,y)。

quiver(x,y,u,v)

上述命令在 x 和 y 中每个对应元素对指定的坐标处绘制矢量作为箭头。

参数

下表列出了箭羽图的不同参数:

x 1D 或 2D 数组,序列。箭头的 x 坐标位置
y 1D 或 2D 数组,序列。箭头的 y 坐标位置
u 1D 或 2D 数组,序列。箭头矢量的 x 分量
v 1D 或 2D 数组,序列。箭头矢量的 y 分量
c 1D 或 2D 数组,序列。箭头的颜色

以下代码绘制了一个简单的箭羽图:

import matplotlib.pyplot as plt
import numpy as np
x,y = np.meshgrid(np.arange(-2, 2, .2), np.arange(-2, 2, .25))
z = x*np.exp(-x**2 - y**2)
v, u = np.gradient(z, .2, .2)
fig, ax = plt.subplots()
q = ax.quiver(x,y,u,v)
plt.show()
Quicker Plot
广告