要使用matplotlib绘制2D数学向量,我们可以采取以下步骤-
使用numpy数组创建矢量坐标。
获取x,y,u和v数据点。
使用图形方法创建一个新图形或激活一个现有图形。
使用gca()方法获取当前轴。
设置轴的x和y极限。
要重绘当前图形,请使用draw()方法。
要显示该图,请使用show()方法。
import numpy as np importmatplotlib.pyplotas plt plt.rcParams["figure.figsize"] = [7.50, 3.50] plt.rcParams["figure.autolayout"] = True soa = np.array([[0, 0, 3, 2], [0, 0, 4, 5], [0, 0, 9, 9]]) X, Y, U, V = zip(*soa) plt.figure() ax = plt.gca() ax.quiver(X, Y, U, V, angles='xy', scale_units='xy', scale=1, color=['red', 'green', 'yellow']) ax.set_xlim([-1, 10]) ax.set_ylim([-1, 10]) plt.draw() plt.show()
输出结果