不管发生什么,都必须遍历整个文件,所以为什么不初始化数组并输入值呢?在
棘手的部分是,如果您想要一个形状为(NX1,NX2,NX3)的数组,但是如果您的x1,x2,x3值是float,那么您必须以某种方式索引数组。也许存在一个数据结构,但是您可以使用类似的def xyz_index((x,y,z),(n1,n2,n3)):
""" return integer indices for x,y,z position
given a constant step """
return tuple(map(int,[x/n1,y/n2,z/n3]))
import numpy as np
NX1,NX2,NX3 = (80, 120, 100)
ns = n1, n2, n3 = (.1,.5,.2)
x1, x2, x3 = np.arange(0,1+n1,n1), np.arange(0,20+n2,n2), np.arange(0,15+n3,n3),
data = np.empty((NX1,NX2,NX3),dtype=[('f',float),('g',float),('h',float)])
with open(filename,'r') as f:
for line in f:
x,y,z,f,g,h = map(float,line.split(', '))
data[xyz_index((x,y,z),ns)] = (f,g,h)
然后您可以按如下方式访问数据:
对于点h-值,使用
^{pr2}$
如果没有['h'],这将返回一个(f,g,h)元组和上面的dtype元组。在
如果没有索引,它将返回一个包含所有h-值的(NX1,NX2,NX3)数组。在
现在我来看看,如果n1, n2, n3总是相同的,那么您可能需要在您的xyz_index函数中定义它们,这样就不必每次都传递ns:
^{3}$