在numpy中将1D数组转换为2D数组

Question

我想通过指定二维数组中的列数，将一维数组转换为二维数组。像这样工作的东西：

> 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”的函数？（我知道您可以像2D数组一样索引1D数组，但在我拥有的代码中这不是一个选项-我需要进行此转换。）

Answer 1

你想^{}数组。

B = np.reshape(A, (-1, 2))

其中-1根据输入数组的大小推断新维度的大小。