Конвертировать 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 вариант в коде у меня есть - мне нужно сделать это преобразование.)