python - 通过 Python 中的迭代在 numpy/scipy 中建立一个数组？

Question

通常，我通过遍历一些数据来构建一个数组，例如:

my_array = []
for n in range(1000):
  # do operation, get value 
  my_array.append(value)
# cast to array
my_array = array(my_array)

我发现我必须先构建一个列表，然后将它(使用“数组”)转换为一个数组。有没有办法解决这些问题？所有这些转换调用使代码变得困惑......我如何迭代地构建“my_array”，它从一开始就是一个数组？

Answer 1

NumPy 提供了一个“fromiter”方法:

def myfunc(n):
    for i in range(n):
        yield i**2


np.fromiter(myfunc(5), dtype=int)

产生

array([ 0,  1,  4,  9, 16])