Конвертировать 1D массив в 2D массив в NumPy

Я хочу преобразовать одномерный массив в двумерный массив, указав количество столбцов в двумерном массиве. Что-то, что будет работать так:

> 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»? (Я понимаю, что вы можете индексировать одномерный массив как двумерный массив, но это неt вариант в коде у меня есть - мне нужно сделать это преобразование.)

Ответы на вопрос(3)

Ваш ответ на вопрос