获取python numpy ndarray的列名

时间:2011-09-26 20:38:00

标签: python arrays numpy names

假设我有一个名为data.txt的数据文件,如下所示:

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]

我想过只是读取数据文件并获取第一行并将其解析为单独的操作,但这似乎不是非常有效或优雅。

1 个答案:

答案 0 :(得分:44)

尝试:

myData.dtype.names

这将返回字段名称的元组。

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