Изменение формы ndarray эффективно

Question

У меня есть ndarray (пример приведен ниже).

A=
[[0.1  1.1 ]
 [0.1  1.3 ]
 [0.25 1.25]
 [0.25 1.45]
 [0.37 1.37]
 [0.35 1.8 ]]

Я хочу изменить его на

B=
[[[0.1  1.1 ] [0.1  1.3 ]]
 [[0.25  1.25 ] [0.25  1.45 ]]
 [[0.37  1.37 ] [0.35  1.38 ]]]

, чтобы сделать

B[0] gives me [[0.1  1.1 ] [0.1  1.3 ]] and B[0][0] gives [0.1  1.1 ]

Answer 1

Следующее должно работать

A.reshape(3,2,2)

Answer 2

Использовать стандарт numpy изменение формы :

B = A.reshape(-1, 2, 2)