在 Python 中计算矢量场的旋度并使用 Matplotlib 进行绘图

matplotlibpythondata visualization

要在 Python 中计算矢量场的旋度并使用 Matplotlib 进行绘图,我们可以使用 quiver() 方法并计算相应的数据。

步骤

  • 设置图形大小并调整子图之间和周围的填充。
  • 使用 figure() 方法创建新图形或激活现有图形。
  • 将 3D 轴添加到图形作为子图排列的一部分。
  • 使用 numpy meshgrid 创建 xyz 数据点。
  • 创建 uvw 数据旋度矢量位置。
  • 使用quiver() 方法获取向量。
  • 关闭轴。
  • 要显示图形,请使用 show() 方法。

示例

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

fig = plt.figure()

ax = fig.add_subplot(projection='3d')

x, y, z = np.meshgrid(np.arange(-0.8, 1, 0.2),
                     np.arange(-0.8, 1, 0.2),
                     np.arange(-0.8, 1, 0.8))

u = 0
v = y**2
w = -2*y*z - y

ax.quiver(x, y, z, u, v, w, length=0.1)
ax.axis('off')

plt.show()

输出


相关文章