python - 生成器在传递给 "numpy.array"时不工作是一个错误吗？

Question

如果将生成器传递给 numpy.array，numpy 不会迭代对象，也不会报错，即使 copy=是的。对数组进行索引的任何尝试都会失败，而且通常会在很晚的时候发生，并且在很远的代码中。

我理解 numpy 想从一开始就知道数组的大小，但这种行为并不好。它应该复制到中间列表或引发异常。

Answer 1

这不是错误，而是正常现象。如果要从迭代器创建数组，请使用 fromiter:

>>> import numpy
>>> a = (i*i for i in range(7))
>>> numpy.array(a)
array(<generator object <genexpr> at 0x10dbc1b40>, dtype=object)
>>> numpy.fromiter(a, dtype=float)
array([  0.,   1.,   4.,   9.,  16.,  25.,  36.])