如何使用数据类型(datetime,float)进行numpy重新排列?

时间:2012-04-05 16:57:59

标签: python numpy recarray

假设我有一些简单的数据

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]]

我想要一个与之对应的numpy记录数组。看起来我应该能够做到这一点:

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

或者

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

或者这个

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

但他们每个人都会返回错误

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

TypeError: expected a readable buffer object

那么假设它甚至可能,我怎么能设置它呢?

1 个答案:

答案 0 :(得分:5)

您可以使用元组而不是记录列表:

>> 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')])