2012-04-05 3 views
5

Digamos que tengo algunos datos simplesCómo hacer una recapitulación numpy con tipos de datos (datetime, float)?

y = [[datetime.datetime(2012,1,1,1,1), 2.1], 
    [datetime.datetime(2012,1,1,1,2), -3.1], 
    [datetime.datetime(2012,1,1,1,3), 0.1]] 

y quiero una matriz registro numpy que le corresponde. Parecería que debería ser capaz de hacer esto:

np.rec.array(y, dtype=[('timestamp', object),('x','f')]) 

o esta

np.rec.array(y, dtype=[('timestamp', '|O8'),('x','f')]) 

o tal vez esto

np.rec.array(y, dtype=[('timestamp', 'V'),('x','f')]) 

Pero cada uno de ellos devuelva un error, ya sea

ValueError: Setting void-array with object members using buffer. 

o

TypeError: expected a readable buffer object 

Entonces, ¿cómo puedo configurar esto, suponiendo que sea posible?

Respuesta

5

Puede usar tuplas en lugar de las listas de los registros:

>> y = [(datetime.datetime(2012,1,1,1,1), 2.1), 
...  (datetime.datetime(2012,1,1,1,2), -3.1), 
...  (datetime.datetime(2012,1,1,1,3), 0.1)] 
>> np.rec.array(y, dtype=[('timestamp', object), ('x','f')]) 
rec.array([(datetime.datetime(2012, 1, 1, 1, 1), 2.0999999046325684), 
      (datetime.datetime(2012, 1, 1, 1, 2), -3.0999999046325684), 
      (datetime.datetime(2012, 1, 1, 1, 3), 0.10000000149011612)], 
dtype=[('timestamp', '|O8'), ('x', '<f4')]) 
+0

santo fuma, es así de simple. ¡Gracias! –

Cuestiones relacionadas