读取列格式的三维模型并显示pyvista

最新推荐文章于 2025-06-10 09:00:03 发布

oceanstonetree

最新推荐文章于 2025-06-10 09:00:03 发布

阅读量1.9k

点赞数

CC 4.0 BY-SA版权

分类专栏： python vtk 可视化文章标签： python

本文链接：https://blog.csdn.net/yanfeng1022/article/details/121644331

可视化同时被 3 个专栏收录

40 篇文章

订阅专栏

python

16 篇文章

订阅专栏

vtk

3 篇文章

订阅专栏

本文介绍了使用Python进行数据加载和处理，通过MyDataSet类实现了一个三维模型的可视化，展示了如何读取数据、创建颜色映射并调整z轴比例。关键步骤包括读取'case5.dat'文件，重构数据并应用facies数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

数据格式如下

前四行为基本信息，读取时跨过去，输出时按k，j，i的循环顺序

#coding=utf-8

import numpy as np
import pyvista as pv
import matplotlib.pyplot as plt
import matplotlib as mpl

ni=129
nj=129
nk=100
class MyDataSet:
	def __init__(self,  fileNameList):
		self._fileNameList=fileNameList
		self.create_colorMap()
		self.p=pv.Plotter()		
		self.p.set_scale(zscale=10)
		#self.p.add_slider_widget(self.set_zscale(),rng=np.range(1,100))
		self.p.add_axes()
		
	def readData(self, fileName):
		dat=np.loadtxt(fileName,skiprows=4)
		dat=dat.transpose()
		self.x = dat[3].reshape(nk,nj,ni)
		self.y = dat[4].reshape(nk,nj,ni)
		self.z = dat[5].reshape(nk,nj,ni)
		self.facies = dat[6].reshape(nk,nj,ni)		
		self.facies[self.facies>7]=7
		
	def create_colorMap(self):
		#colors= ['blue','cyan','green','pink','magenta','purple','gold','red']
		colors= ['blue','green','pink','magenta','orange','yellow','red']
		self.discmap = mpl.colors.ListedColormap(colors)
 
	def set_zscale(self, _zscale):
		self.p.set_scale(zscale=_zscale)
 
	def showModel(self):	
		grid = pv.StructuredGrid(self.x, self.y, self.z)
		#grid.point_arrays["values"]=self.facies.flatten(order="F")
		grid.cell_arrays["values"]=self.facies[0:nk-1, 0:nj-1, 0:ni-1].flatten(order="F")
		self.p.add_mesh(grid, cmap=self.discmap,  show_scalar_bar=True)
		self.p.show()

if __name__=="__main__":
	fileNameList=[]
	myDataSet = MyDataSet(fileNameList)
	myDataSet.readData('./case5.dat')
	myDataSet.showModel()

效果如下