將numpy的flatten_dtype與具有標題的結構化dtypes結合使用

Question

我通常不會在這些論壇上發佈問題，但我已遍佈整個地方搜索，並且我沒有發現任何有關此問題的信息。

我正在使用結構化數組來存儲實驗數據。我使用標題來存儲有關我的字段的信息，在這種情況下是度量單位。當我在我的D型呼叫numpy.lib.io.flatten_dtype（），我得到：

ValueError: too many values to unpack 
File "c:\Python25\Lib\site-packages\numpy\lib\_iotools.py", line 78, in flatten_dtype 
    (typ, _) = ndtype.fields[field] 

我不會很在意，只是numpy.genfromtxt（）調用numpy.lib.io.flatten_dtype（） ，我需要能夠從文本文件中導入我的數據。

我想知道我是否做錯了什麼。 flatten_dtype（）不是用來支持標題的嗎？ genfromtxt（）有沒有解決方法？

這裏是我的代碼片段：

import numpy 
fname = "C:\\Somefile.txt" 
dtype = numpy.dtype([(("Amps","Current"),"f8"),(("Volts","Voltage"),"f8")]) 
myarray = numpy.genfromtxt(fname,dtype) 

Answer 1

這裏是一個可能的解決方法：

由於您的自定義dtype會導致一個問題，提供而不是扁平的D型：

In [77]: arr=np.genfromtxt('a',dtype='f8,f8') 

In [78]: arr 
Out[78]: 
array([(1.0, 2.0), (3.0, 4.0)], 
     dtype=[('f0', '<f8'), ('f1', '<f8')]) 

然後請使用astype轉換爲您想要的dtype：

In [79]: arr=np.genfromtxt('a',dtype='f8,f8').astype(dtype) 

In [80]: arr 
Out[80]: 
array([(1.0, 2.0), (3.0, 4.0)], 
     dtype=[(('Amps', 'Current'), '<f8'), (('Volts', 'Voltage'), '<f8')]) 

編輯：另一種方法是猴子補丁numpy.lib.io.flatten_dtype：

import numpy 
import numpy.lib.io 
def flatten_dtype(ndtype): 
    """ 
    Unpack a structured data-type. 

    """ 
    names = ndtype.names 
    if names is None: 
     return [ndtype] 
    else: 
     types = [] 
     for field in names: 
      typ_fields = ndtype.fields[field] 
      flat_dt = flatten_dtype(typ_fields[0]) 
      types.extend(flat_dt) 
     return types 
numpy.lib.io.flatten_dtype=flatten_dtype