я хочу вычесть 3D-массив для цикла

Question

Я хочу вычесть предыдущее значение из приведенного значения, но я застрял в трехмерном массиве?

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 в 3d (72,80,97), но он показывает 2d массив.

Answer 1

вы переназначили diffrain за каждый j. Вместо этого сделайте

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)