重塑阵列

【问题标题】：Reshaping array重塑阵列
【发布时间】：2021-08-31 01:20:27
【问题描述】：

我需要转换数组来构建混淆矩阵

如何转换这个数组

array([5, 2, 8, ..., 1, 2, 2])

所以它看起来像这样？

array([[5],
   [2],
   [8],
   ...,
   [1],
   [2],
   [2]])

【问题讨论】：

my_arr.shape = -1, 1 ?
np.vstack(array) - numpy.org/doc/stable/reference/generated/numpy.vstack.html

标签： python numpy

【解决方案1】：

>>> array=[5, 2, 8,  1, 2, 2]
>>> [ [x] for x in array]
[[5], [2], [8], [1], [2], [2]]

可以通过列表理解来完成

【讨论】：

【解决方案2】：

In [1009]: arr = np.arange(4)
In [1010]: arr
Out[1010]: array([0, 1, 2, 3])

正如你的标题所说：

In [1011]: arr1 = arr.reshape(4,1)
In [1012]: arr1
Out[1012]: 
array([[0],
       [1],
       [2],
       [3]])

另一种添加维度的惯用方式：

In [1013]: arr1 = arr[:,None]     # or with np.newaxis
In [1014]: arr1
Out[1014]: 
array([[0],
       [1],
       [2],
       [3]])
In [1015]: arr.shape
Out[1015]: (4,)
In [1016]: arr1.shape
Out[1016]: (4, 1)

【讨论】：