Получить имена столбцов Python NumPy ndarray - PullRequest
Новая
       8

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

42 голосов
/ 27 сентября 2011

Допустим, у меня есть файл данных с именем data.txt, который выглядит следующим образом: 

TIME FX FY FZ
0    10 5  6
1    2  4  7
2    5  2  6
...

В Python Run: 

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 Ответ

46 голосов
/ 27 сентября 2011

Попробуйте: 

myData.dtype.names

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

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