获取Python numpy ndarray的列名

Question

假设我有一个数据文件，叫做 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]

我考虑过直接读取数据文件，获取第一行，然后单独处理它，但这样做似乎既不高效也不优雅。

Answer 1

试试这个：

myData.dtype.names

这样做会返回一个包含字段名称的元组。

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