2011-09-26 980 views
34

比方說,我有一個名爲data.txt一個數據文件,它看起來像:獲取一個python numpy的ndarray的列名

TIME FX FY FZ 
0 10 5 6 
1 2 4 7 
2 5 2 6 
... 

在Python中運行:

import numpy as np 

myData = np.genfromtxt("data.txt", names=True) 

>>> print myData["TIME"] 
[0, 1, 2] 

在頂端的名字我數據文件會有所不同,所以我想要做的是找出數據文件中的數組名稱。所以,我想是這樣的:

>>> print myData.names 
[TIME, F0, F1, F2] 

我想到了剛剛在數據文件中讀取並獲得第一行,並解析它作爲一個單獨的操作,但是這似乎並不非常有效或優雅。

回答

35

嘗試:

myData.dtype.names 

這將返回字段名稱的元組。

In [10]: myData.dtype.names 
Out[10]: ('TIME', 'FX', 'FY', 'FZ')