在 Matplotlib 的 3D 图形中为向量添加箭头


要在 3D matplotlb 的图形中绘制箭头头的向量,我们可以采取以下步骤 -

  • 创建一个二维数组,其中 x、y、z、u、v和w 是箭头位置的坐标以及箭头向量方向的组件。

  • 使用 figure() 方法创建一个新图形或激活一个现有图形。

  • 使用 add_subplot() 方法将 '~.axes.Axes' 作为子图排列的一部分添加到图形中。

  • 使用 quiver() 方法绘制箭头 3D 域。

  • 使用 ylim, xlim, zlim, 限制各个轴的范围。

  • 设置图形的标题。

  • 要显示图形,请使用 show() 方法。

示例

import matplotlib.pyplot as plt
import numpy as np
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
soa = np.array([[0, 0, 1, 1, -2, 0], [0, 0, 2, 1, 1, 0], [0, 0, 3, 2, 1, 0], [0, 0, 4, 0.5, 0.7, 0]])
X, Y, Z, U, V, W = zip(*soa)
fig = plt.figure()
ax = fig.add_subplot(projection='3d')
ax.quiver(X, Y, Z, U, V, W, color='red')
ax.set_xlim([-1, 0.5])
ax.set_ylim([-1, 1.5])
ax.set_zlim([-1, 8])
ax.set_title("Vectors")
plt.show()

输出

更新时间: 07-5-2021

2 千次+ 浏览

开启您的事业

完成课程认证

立即开始
广告