用标量函数填充现有numpy数组

Question

我有一个一维numpy数组，需要用数据重新填充。给定索引处的数据由函数foo计算

我现在拥有的是

for in in range(n):
    arr[i] = foo(i, state)

这是代码的性能瓶颈。我考虑过numpy.fromiter，但它每次都返回一个新数组。我宁愿重用数组来节省空间和时间。你知道吗