Подтвердить что ты не робот

Получить имена столбцов python numpy ndarray

Скажем, у меня есть файл данных с именем 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]

Я думал просто прочитать в файле данных и получить первую строку и проанализировать ее как отдельную операцию, но это не кажется очень эффективным или элегантным.

4b9b3361

Ответ 1

Try:

myData.dtype.names

Это вернет кортеж имен полей.

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