2011-09-26 11 views
34

Załóżmy, że mam plik danych o nazwie data.txt, który wygląda tak:Get nazwy kolumn o ndarray python numpy

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

w Pythonie Run:

import numpy as np 

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

>>> print myData["TIME"] 
[0, 1, 2] 

Nazwy na szczycie mojej plik danych będzie się różnić, więc chciałbym się dowiedzieć, jakie są nazwy moich tablic w pliku danych. Więc chciałbym coś takiego:

>>> print myData.names 
[TIME, F0, F1, F2] 

myślałem o tylko do odczytu w pliku danych i uzyskać pierwszą linię i przetwarza je jako oddzielnej operacji, ale to nie wydaje się bardzo skuteczny i elegancki.

Odpowiedz

35

Spróbuj:

myData.dtype.names 

ten powróci krotki nazw terenowych.

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