如何使用 matplotlib 绘制二维 FEM 结果？

有限元方法 (FEM) 用于各种任务，例如对不同材料类型建模、测试复杂几何形状、对作用在设计的小区域内的局部效应进行可视化。它基本上将一个较大的空间域分解成称为“有限元”的简单部分。然后将对这些有限元建模的简单方程收集到一个较大的方程组中，以对整个域进行建模。

要使用 matplotlib 绘制二维 FEM 结果，我们可以采取以下步骤 −

  • 设置图形大小并调整子图之间和周围的内边距。
  • 使用 numpy 创建节点、元素和节点值数据点。
  • 转置节点的数据点。
  • 使用 tricontourf() 创建一个 3D 填充等值线图。
  • 要显示图形，请使用 show() 方法。

示例

import numpy as np
import matplotlib.pyplot as plt

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

nodes = np.array([
   [0.0, 0.0],
   [1.0, 0.0],
   [2.0, 0.5],
   [0.0, 1.0],
   [1.0, 1.0],
   [1.7, 1.3],
   [1.0, 1.7]])

elements = np.array([
   [1, 2, 5],
   [5, 4, 1],
   [2, 3, 6],
   [6, 5, 2],
   [4, 5, 7],
   [5, 6, 7]])

values = [1, 2, 1, 2, 7, 4, 5]

x, y = nodes.T

plt.tricontourf(x, y, elements - 1, values, 12, cmap='copper')

plt.show()

输出

它将产生以下输出

更新于: 2021-09-21

673 次浏览

开启您的 职业生涯

完成课程，获得认证

学习
广告