2012-05-30 23 views
8

¿Cómo puedo hacer que genfromtxt me devuelva su lista de nombres de columnas que se recuperaron automáticamente por los nombres = Verdadero? Cuando lo hagoNumpy genfromtxt Nombres de columnas

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

Imprime los valores de columna completos para col1. Pero necesito atravesar todos los nombres de columna; ¿Cómo puedo hacer eso? Intenté data.keys() y varios otros métodos, pero todo lo que se devuelve por genfromtxt no parece ser un objeto compatible con el diccionario. Supongo que podría pasar la lista de nombres de columnas, pero esto no me será posible a largo plazo.

¿Alguna idea?

Respuesta

13

genfromtxt devuelve un numpy.ndarray.

que pueda obtener el tipo de datos con

data.dtype 

o sólo los nombres con

data.dtype.names 

que es una tupla puede repetir una y acceder a las columnas que desea.

Cuestiones relacionadas