python - 如何自动生成 numpy dtype？

Question

我有多个数组，数组名称为

   Level1
   Level2
   Level3
   .
   .

等等。每个数组有 4 列和任意数量的行。列名的形式为

 AP%i BP%i AS%i BS%i

其中%i 对应于数组名称中的相应索引（例如Level1 -> AP01 BP01 AS01 BS01）。如何创建具有正确列名的此类数组的 dtype，其中列名是变量？

Answer 1

您可以使用这样的东西来动态生成所需的 dtypes：

for i in xrange(1, N+1): # N is number of arrays
    arr = globals()['Level%i' % i] # this gets the Level<X> value for each i
    arr.dtype = [('AP%02i' % i,float), ('BP%02i' % i, float), ('AS%02i' % i, float), ('BS%02i' % i, float)]
# example
print Level1[0]['AP01']

请记住根据您实际拥有的数据类型调整 dtype 中的类型。