我想在for循环中减去3D数组

Question

我想用当前值减去上一个值，但我在3d数组中卡住了

print(rain.shape) #(73,80,97)
for j in range(1,72):
      diffrain=np.array(rain[j+1,:,:]-rain[j,:,:])

print(diffrain.shape) #(80,97) only 2d array what about j, here j.shape=73 after subtrcation the value of j will be 72.

我想要三维的diffrain.shape（72,80,97），但它显示的是二维数组

Answer 1

您为每个j重新分配了diffrain。相反，你应该这样做

diffrain = []
for j in range(1,72):
      diffrain.append(rain[j+1,:,:]-rain[j,:,:])

diffrain = np.stack(diffrain)
print(diffrain.shape) # now (72,80,97)