将2D数组函数返回值分配给新的numpy数组

Question

我是python的新手，有一个愚蠢的问题。 我抬头寻找答案，找不到答案，所以在这里（StackOverflow上的第一个问题，如果有任何错误，请原谅）。

我有一个功能完美的函数（functions.Reservoir）。 它需要一个复杂的numpy 2D数组和一个浮点值作为参数。 它返回另一个复杂的numpy 2D数组。 我尝试将所有这些返回值保存在一个新的numpy数组中。 问题是第一个打印命令为我提供了正确的值，这意味着我的功能正常运行。 但是在for循环之外，当我打印l变量时，所有元素都具有相同的值，即最后一个元素应该具有的值。

你能告诉我我做错了吗？

您可以帮助我的另一件事是一种将这些返回值保存为3D numpy数组的方法。 我不想使用vstack，因为有时我的循环会迭代4000次，而vstack会大大降低它的速度。 我现在要做的是将numpy数组另存为对象，然后在此迭代器中创建一个新数组。 如果有更好的方法，请告诉我。 提前致谢

    k= np.zeros((400,3), dtype=np.complex128)
    l= np.empty(noPoints1, dtype=object)
    for i in  xrange(noPoints1):
      l[i] = functions.Reservoir(k, data[ OFFSET1+ Discard +i] )
      print l[i]
      k=l[i]
    print l

更新资料

这里是我的代码的更多信息：

def determ(k1, k2,  d): 
    dkdt = np.zeros_like(k1)

    EA1 = np.absolute(k1[0])**2
    EA2 = np.absolute(k1[1])**2

    gain1= g1*(k1[2]-N0)/(1+s*EA1)
    gain2= g2*(k1[2]-N0)/(1+s*EA2)

    dkdt[0]= 0.5 * (1+ A*1j) * (gain1 - pd) * k1[0] 

    dkdt[1]= 0.5 * (1+ A*1j) * (gain2 - pd) * k1[1] 

    dkdt[2]= Stroom/e - k1[2]*ed - EA1*gain1 - EA2*gain2   
    return dkdt

def Reservoir(y,d): 
    q= y.shape[0]               
    mdata= 0.5*np.pi*d*maskvector    #Maskvector 1D with q elements

    for n in xrange(q):  
      yn = y[n,:]               # Predictor

      if n+1<q:
        ytd= y[n+1,:]           # Predictor Delayed Field strength
      else:
        ytd= y[n+1-q,:]
      #See y as the state of a circular reservoir. If you reach the end at q, then it directly connects back at the start.

      fn = determ(k1=yn, k2=ytd, d=mdata[n]) 

      ybar = yn + fn*step           # Corrector
      if n+2<q:
        ytdbar= y[n+2,:]            # Corrector Delayed Field strength  
      else:
        ytdbar= y[n+2-q,:]

      fnbar = determ(k1= ybar, k2=ytdbar, d=mdata[n])

      if n+1<q:
        y[n+1] = yn + 0.5*(fn + fnbar)*step 
      else:
        y[n+1-q] = yn + 0.5*(fn + fnbar)*step 
    return (y)

PS。 我尝试了HYRY（ 从返回NumPy数组的循环中收集结果）的答案，但是对于所有元素，我仍然得到相同的值。

Answer 1

看来您总是在同一阵列上工作。 在k=l[i].copy()更改将确保在每个步骤上都对新数组起作用。

Reservoir的return y不会按照docs中的说明创建新数组：

请注意，数组切片不会复制内部数组数据，但还会产生原始数据的新视图，