2012-05-30 31 views
8

Come posso avere genfromtxt per restituirmi l'elenco dei nomi di colonna che sono stati recuperati automaticamente da names = True? Quando faccioNumpy genfromtxt Nome colonna

data = np.genfromtxt("test.csv",names=True,delimiter=",",dtype=None) 
print data['col1'] 

Stampa i valori dell'intera colonna per col1. Ma ho bisogno di attraversare tutti i nomi delle colonne; Come lo posso fare? Ho provato data.keys() e vari altri metodi, ma tutto ciò che viene restituito da genfromtxt non sembra essere un oggetto compatibile con il dizionario. Credo di poter passare da solo l'elenco dei nomi delle colonne, ma a lungo andare questo non sarà più sostenibile per me.

Qualche idea?

risposta

13

genfromtxt restituisce un numpy.ndarray.

è possibile ottenere il tipo di dati con

data.dtype 

o solo i nomi con

data.dtype.names 

che è una tupla è possibile iterare ed accedere alle colonne come si desidera.

Problemi correlati