tutorial - Obtener los nombres de las columnas de un python numpy ndarray

numpy tutorial español pdf

Question

Digamos que tengo un archivo de datos llamado data.txt que se ve así:

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

En Python Run:

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 archivo de datos variarán, entonces lo que me gustaría hacer es averiguar cuáles son los nombres de mis matrices en el archivo de datos. Entonces me gustaría algo como:

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

Pensé simplemente leer en el archivo de datos y obtener la primera línea y analizarla como una operación separada, pero eso no parece muy eficiente o elegante.

Answer 1

Tratar:

myData.dtype.names

Esto devolverá una tupla de los nombres de campo.

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