使用pyevtk导出结构化VTK网格以供后处理

pyevtk简介

在计算流体力学CFD中，通常需要处理三维网格数据，为了可视化，需要将其输出。本文介绍使用python的pyevtk库输出结构化网格，以供paraview进一步后处理。

代码

# **************************************************************
# * Example of how to use the high level gridToVTK function.   *
# * This example shows how to export a structured grid.        *
# **************************************************************

from pyevtk.hl import gridToVTK
import numpy as np

# 尺寸参数
lx, ly, lz = 3.0, 4.0, 5.0 #立方体的长宽高
nx_points, ny_points, nz_points = 3, 4, 5 #边上的节点数
n_points= nx_points*ny_points*nz_points #节点总数
nx_cells, ny_cells, nz_cells = nx_points-1, ny_points-1, nz_points-1 #边上的单元数
n_cells= nx_cells*ny_cells*nz_cells #单元总数
dx,dy,dz=lx/nx_cells,ly/ny_cells,lz/nz_cells #节点间距

# 节点坐标
x_vector = np.linspace(0,lx,nx_points) #x方向节点分布，向量
y_vector = np.linspace(0,ly,ny_points) #y方向节点分布，向量
z_vector = np.linspace(0,lz,nz_points) #z方向节点分布，向量
x_field,y_field,z_field=np.meshgrid(x_vector,y_vector,z_vector, indexing='ij',sparse=False) #x场，y场，z场张量，注意：indexing务必选择'ij'

# 物理量
pressure = np.random.rand(n_cells).reshape((nx_cells, ny_cells, nz_cells)) #随机压力场（存储在体心）
temp = np.random.rand(n_points).reshape(