如何将Python字典对象转换为numpy数组

Question

我有python dict对象，键为datetime.date对象，值为元组对象：

>>> data_dict
{datetime.date(2006, 1, 1): (5, 3),
 datetime.date(2006, 1, 2): (8, 8),
 datetime.date(2006, 1, 3): (8, 5),
 datetime.date(2006, 1, 4): (3, 3),
 datetime.date(2006, 1, 5): (3, 3),
 datetime.date(2006, 1, 6): (4, 3),
...

我希望以这种格式将其转换为numpy数组对象：

dtype([('date', '|O4'), ('high', '<i1'), ('low', '<i1')])

这样我就可以将它存储在磁盘上，然后再使用它，并在numpy中学习matplotlib ......

事实上，在查看这个matplotlib示例后，我想要使用这种格式：http://matplotlib.sourceforge.net/users/recipes.html但是找不到如何到达那里的方法。

Answer 1

以下将会这样做：

arr = np.array([(k,)+v for k,v in data_dict.iteritems()], \
         dtype=[('date', '|O4'), ('high', '<f8'), ('low', '<f8')])

如果您希望将arr用作recarray，则可以使用：

arr = arr.view(np.recarray)

这样您就可以按名称引用字段，例如arr.date。