实验室导师安排学习流场三维可视化方向的相关知识。
特此记录学习的一些收获
import pyvista as pv
import numpy as np
data = np.genfromtxt('1.csv', delimiter=',', skip_header=1)
x = data[:, 0]
y = data[:, 1]
z = data[:, 2]
result = data[:,-2]
points = pv.PolyData(np.column_stack((x, y, z)))
# 将结果添加到点云对象中作为颜色属性
points['Color'] = result
result_min = np.min(result)
result_max = np.max(result)
rng = [result_min, result_max]
# rng = [result_min + 0.1 * (result_max - result_min), result_max]
# 使用plot函数可视化点云对象,并使用颜色映射将结果映射到颜色
pv.plot(points, scalars='Color', cmap='jet', rng=rng)
代码实现流场以点的形式进行三维可视化,并加入颜色映射。
接下来准备以流线、和等势面的形式对流场数据可视化