获取python numpy ndarray的列名

2024-05-16 18:07:30 发布

您现在位置:Python中文网/ 问答频道 /正文

假设我有一个名为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]

我只想读入数据文件,得到第一行并将其解析为一个单独的操作,但这似乎不是很有效或优雅。


Tags: importnumpytxt名称datatimenames数据文件