如何从生成器构建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)