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

10 浏览
0 Comments

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

我想通过指定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"的函数？（我知道可以像二维数组那样索引一维数组，但在我的代码中没有这个选项 - 我需要进行转换。）

0