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

Question

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

计算

我现在拥有的是像

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

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