Rearranging a numpy array to fit its grid cell

Question

I coded a YOLO model from scratch and have a numpy array which looks like this:

[
 [[1 0 1 0.4 0.3 0.2 0.1]
 [1 1 0 0.2 0.3 0.4 0.5]
 [0 0 0 0 0 0 0]]
 ...]