Converta uma matriz 1D em uma matriz 2D em numpy
Eu quero converter uma matriz unidimensional em uma matriz bidimensional, especificando o número de colunas na matriz 2D. Algo que funcionaria assim:
> 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]])
O numpy tem uma função que funciona como a minha função "vec2matrix"? (Eu entendo que você pode indexar uma matriz 1D como uma matriz 2D, mas isso não é uma opção no código que eu tenho - eu preciso fazer essa conversão.)