Я хочу перетворити одновимірний масив у двовимірний масив, вказавши кількість стовпців у двовимірному масиві. Щось, що спрацювало б так:
> 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]])
Чи має numpy функцію, яка працює як моя створена функція "vec2matrix"? (Я розумію, що ви можете індексувати 1D масив, як 2D масив, але це не варіант у коді, який у мене є - мені потрібно здійснити це перетворення.)