如何从生成器构建numpy数组？

Question

如何从生成器对象构建numpy数组？

让我说明一下这个问题：

>>> import numpy

>>> def gimme():

...   for x in xrange(10):

...     yield x

...

>>> gimme()

<generator object at 0x28a1758>

>>> list(gimme())

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

>>> numpy.array(xrange(10))

array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

>>> numpy.array(gimme())

array(<generator object at 0x28a1758>, dtype=object)

>>> numpy.array(list(gimme()))

array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

在这种情况下，gimme（）是生成器，我要将其输出转换为数组。但是，数组构造函数不会在生成器上进行迭代，它只是存储生成器本身。我想要的行为是来自numpy.array（list（gimme（））），但是我不想支付同时拥有中间列表和最终数组的内存开销。有没有更节省空间的方法？

Answer 1

这个stackoverflow结果背后的一个Google，我发现有一个numpy.fromiter(data, dtype, count)。默认值count=-1从可迭代对象中获取所有元素。它需要dtype明确设置。就我而言，这可行：numpy.fromiter(something.generate(from_this_input), float)