在循环中附加二维数组

时间:2019-04-18 15:52:25

标签: python numpy multidimensional-array

我想将某些值存储在2D数组中。在下面的代码中。我希望sTtotal。当内循环运行时,值将存储在行中,然后在外循环递增时存储在下一列中。

class pricing_lookback:
  def __init__(self,spot,rate,sigma,time,sims,steps):
    self.spot = spot
    self.rate = rate
    self.sigma = sigma
    self.time = time
    self.sims = sims
    self.steps = steps
    self.dt = self.time/self.steps

  def call_floatingstrike(self):

      simulationS = np.array([])
      simulationSt = np.array([])
      call2 = np.array([])
      total = np.empty(shape=[self.steps, self.sims])
      for j in range(self.sims):
          sT = self.spot
          pathwiseminS = np.array([])
          for i in range(self.steps):
              phi= np.random.normal()
              sT *= np.exp((self.rate-0.5*self.sigma*self.sigma)*self.dt + self.sigma*phi*np.sqrt(self.dt))
              pathwiseminS = np.append(pathwiseminS, sT)
              np.append(total,[[j,sT]])###This should store values in rows of j column
              #print (pathwiseminS)
          #tst1 = np.append(tst1, pathwiseminS[1])
          call2 = np.append(call2, max(pathwiseminS[self.steps-1]-self.spot,0))
          #print (pathwiseminS[self.steps-1])
          #print(call2)
          simulationSt = np.append(simulationSt,pathwiseminS[self.steps-1])
          simulationS =  np.append(simulationS,min(pathwiseminS))
      call = max(np.average(simulationSt) - np.average(simulationS),0)
      return call, total#,call2, 

1 个答案:

答案 0 :(得分:0)

以下是我想尝试做的一个简单示例:

for i in range(5):
    row = np.random.rand(5,)
    if i == 0:
        my_array = row
    else:
        my_array = np.vstack((my_array, row))
    print(row)

但是,这在内存方面不是很有效,尤其是在处理大型数组时,因为这必须在每个循环上分配新的内存。最好预先分配一个空数组,然后尽可能填充它。

要回答有关如何添加列的问题,将是这样的:

import numpy as np

x = np.random.rand(5, 4)
column_to_append = np.random.rand(5,)
np.insert(x, x.shape[1], column_to_append, axis=1)

同样,这不是有效的内存,应尽可能避免。预分配要好得多。