Konwertuj tablicę 1D na tablicę 2D w numpy

Chcę przekształcić tablicę jednowymiarową w tablicę dwuwymiarową, określając liczbę kolumn w tablicy 2D. Coś, co zadziałałoby w ten sposób:

> import numpy as np
> A = np.array([1,2,3,4,5,6])
> B = vec2matrix(A,ncol=2)
> B
array([[1, 2],
       [3, 4],
       [5, 6]])

Czy numpy ma funkcję, która działa jak moja wymyślona funkcja „vec2matrix”? (Rozumiem, że możesz indeksować tablicę 1D jak tablicę 2D, ale to nie jest opcja w kodzie, który mam - muszę dokonać tej konwersji).

questionAnswers(3)

yourAnswerToTheQuestion