Selecting a column vector from a matrix in Python

Question

I would like to index a column vector in a matrix in Python/numpy and have it returned as a column vector and not a 1D array.

x = np.array([[1,2],[3,4]])
x[:         


        

Answer 1

Few options -

x[:,[1]]
x[:,None,1]
x[:,1,None]
x[:,1][:,None]
x[:,1].reshape(-1,1)
x[None,:,1].T