python将输入值转化为元组_将坐标元组信息转换为numpy数组

不管发生什么，都必须遍历整个文件，所以为什么不初始化数组并输入值呢？在

棘手的部分是，如果您想要一个形状为(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}$