获取python numpy ndarray的列名

mcr*_*gun 47 python arrays numpy names

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

TIME FX FY FZ
0    10 5  6
1    2  4  7
2    5  2  6
...
Run Code Online (Sandbox Code Playgroud)

在python运行中:

import numpy as np

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

>>> print myData["TIME"]
[0, 1, 2]
Run Code Online (Sandbox Code Playgroud)

我的数据文件顶部的名称会有所不同,所以我想要做的是找出数据文件中我的数组的名称.所以我想要像:

>>> print myData.names
[TIME, F0, F1, F2]
Run Code Online (Sandbox Code Playgroud)

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

Jos*_*del 51

尝试:

myData.dtype.names
Run Code Online (Sandbox Code Playgroud)

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

In [10]: myData.dtype.names
Out[10]: ('TIME', 'FX', 'FY', 'FZ')
Run Code Online (Sandbox Code Playgroud)