Numpy 數組，但每個 1 是一個元素

Question

我知道np.ones(3)給出了以下 np 數組： [1, 1, 1]在 numpy 中是否有一個命令給出以下內容： [[1], [1], [1]] ？

或者，我如何將第一個數組（不僅僅是 3 的大小，而是一個 n 大小的np.ones()數組）轉換為第二個數組的形式。

Answer 1

重塑在這里聽起來像是一個簡單的通用解決方案：

np.ones(3).reshape(-1, 1)

雖然你總是可以從這個開始：

np.ones((3, 1))

這些做同樣的事情

Answer 2

import numpy as np
one=np.ones(3)
one.reshape(3,1,1)