Numpy 数组，但每个 1 是一个元素

Question

I know that np.ones(3) gives the following np array: [1, 1, 1] is there a command in numpy that gives the following: [[1], [1], [1]] ?

Alternatively, how do I turn the first array (not just for the size of 3, but for an n sized np.ones() array) into the form of the second one.

Answer 1

Reshape sounds like an easy general solution here:

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

Although you can always start with this:

np.ones((3, 1))

These do the same thing

Answer 2

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