web-dev-qa-db-fra.com

Récupère les noms de colonne d'un python numpy ndarray

Disons que j'ai un fichier de données appelé data.txt qui ressemble à:

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

Dans python run:

import numpy as np

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

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

Les noms en haut de mon fichier de données varient, donc ce que je voudrais faire est de savoir quels sont les noms de mes tableaux dans le fichier de données. Je voudrais donc quelque chose comme:

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

J'ai pensé à lire le fichier de données et à obtenir la première ligne et à l'analyser comme une opération distincte, mais cela ne semble pas très efficace ou élégant.

44
mcragun

Essayer:

myData.dtype.names

Cela renverra un Tuple des noms de champs.

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