如何更改 numpy 中的数组形状？

Question

如果我创建一个数组 X = np.random.rand(D, 1) 它的形状是 (3,1) ：

 [[ 0.31215124]
 [ 0.84270715]
 [ 0.41846041]]

如果我创建自己的数组 A = np.array([0,1,2]) 那么它的形状是 (1,3) 看起来像

[0 1 2]

我怎样才能在我的阵列上强制形状 (3, 1) A ？

原文由 theQman 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

您可以将形状元组直接分配给 numpy.ndarray.shape 。

 A.shape = (3,1)

截至 2022 年，文档状态：

不鼓励设置 arr.shape，将来可能会弃用。使用 ndarray.reshape 是首选方法。

目前最好的解决方案是

A = np.reshape(A, (3,1))

原文由 alec_djinn 发布，翻译遵循 CC BY-SA 4.0 许可协议