二维初始化numpy数组

Question

因此，如果我想要一个numpy数组[0,1,2,3,4,5,6]（或其他任何具有7个元素的矢量），其尺寸为（7,1）而不是（7 ，），在我创建它时仍可以执行此操作，而不是编写 t=np.expand_dims(np.array(range(7)),axis=1)吗？

Answer 1

我能想到的最好的方法是

i = range(7)

j = np.array(i)[:, None]

一种稍短的方法是：

j = np.atleast_2d(i).T

Answer 2

只需换位

x = np.array([range(7)]).T