У меня есть два массива numpy sample и r.
sample - вложенный массив, а r - плоский массив (1-D).
Я хочу изменить массив массивов r, похожий на форму массива sample.
import numpy as np
sample = np.array([[[[1,0,0,1],[0,0.8,0.7,1]],[[2,2,0,1],[0,0.8,0.7,1]]],[[[1,0,0],[0,0.8,0.7]],[[1,1,0],[0,0.25,0.45]]],[[[0,1],[0,4]]]])
r = np.array([2,0,0,2,0,0.81,0.71,11,2,2,0,1,0,0.8,0.7,1,1,0,0,0,0.8,0.7,1,1,0,0,0.25,0.45,0,10,0,40])
желаемый массив:
r_reshaped = np.array([[[[2,0,0,2],[0,0.81,0.71,11]],[[2,2,0,1],[0,0.8,0.7,1]]],[[[1,0,0],[0,0.8,0.7]],[[1,1,0],[0,0.25,0.45]]],[[[0,10],[0,40]]]])