Matplotlib 中的箭袋图

Python 5天前 10

本教程教你如何使用 Matplotlib 制作箭袋图。

使用 Matplotlib 创建箭袋图

我们需要为 matplotlib.pyplot.quiver 函数提供四个参数来制作图表。X、Y 轴和 X、Y 矢量方向。

x 坐标是坐标数组的第一项，第二项是 Y 坐标。下一个数组也是如此，它代表向量。

代码：

import matplotlib.pyplot as plt
coordinates = [0, 0]
vector = [1, 2]
plt.quiver(coordinates[0], coordinates[1], vector[0], vector[1])

输出：

Matplotlib 中的箭袋图

我们还可以制作多个指向不同方向的箭头。请记住，所有数组都需要具有相同数量的项目或长度。

代码：

import matplotlib.pyplot as plt
from random import randrange
x = [x for x in range(10)]
y = [y for y in range(10)]
vec_x = [randrange(-2, 2) for x in range(10)]
vec_y = [randrange(-2, 2) for x in range(10)]
plt.quiver(x, y, vec_x, vec_y)

输出：

Matplotlib 中的箭袋图

完整代码：

import matplotlib.pyplot as plt
from random import randrange
x = [x for x in range(10)]
y = [y for y in range(10)]
vec_x = [randrange(-2, 2) for x in range(10)]
vec_y = [randrange(-2, 2) for x in range(10)]
plt.quiver(x, y, vec_x, vec_y)