在numpy中将一维数组转换为二维数组

Question

我想通过指定2D数组中的列数将一维数组转换为二维数组。可能会像这样工作：

> 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

您要reshape阵列。

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

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