C[..., (0), (0)] = A[:, (0)][:, (None), (None)]
C[..., (1), (0)] = A[:, 3:][:, (None), :]
C[..., :, (1)] = A[:, 1:3][:, :, (None), (None)]
Создайте новый массив numpy на основе условий, установленных в массиве numpy