2011-09-26 29 views
34

Digamos que tengo un archivo de datos llamado data.txt que se parece a:Obtener los nombres de columna de un ndarray pitón numpy

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

En pitón de ejecución:

import numpy as np 

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

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

Los nombres en la parte superior de mi el archivo de datos variará, entonces lo que me gustaría hacer es averiguar cuáles son los nombres de mis matrices en el archivo de datos. Así que me gustaría algo así como:

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

pensaba sólo para leer en el archivo de datos y obtener la primera línea y analizarlo como una operación separada, pero que no parece muy eficiente o elegante.

Respuesta

35

Probar:

myData.dtype.names 

Esto devolverá una tupla de los nombres de campo.

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