我正在尝试使用python创建3D数据集(我不想使用现有数据集)。 输出将是vtk结构化的数据格式。数据集的大小是用户输入(当前我正在尝试使用216 * 216)在-50.0到50.0的范围内。 我正在通过圆公式计算每个网格位置的标量值 sqrt(xx + yy + zz) 我已经写了下面的代码,但由于没有文件正在生成,而且似乎不确定,我什至不敢尝试
import vtk
import math
x=0
y=0
z=0
print("enter size of the dataset ")
limit=int(input())
for i in range(0,limit):
for j in range(0,limit):
for k in range(0,limit):
x=i
y=j
z=k
vol=math.sqrt(pow(x,2)+pow(y,2)+pow(z,2))
w=vtk.vtkDataSetWriter()
w.SetFileName('C:/Users/prakhyati/datafile1.vtk')
w.Update()
w.Write()