ndarray.reshape()

Меняет форму, не меняя данные (число элементов должно совпадать). Одно из измерений можно задать как -1 — оно вычислится автоматически.

import numpy as np
a = np.arange(6)
print(a.reshape(2, 3))   # [[0 1 2]
                         #  [3 4 5]]
print(a.reshape(3, -1))  # [[0 1]
                         #  [2 3]
                         #  [4 5]]