从Python中的矩阵中选择列向量

Question

我想在Python / numpy的矩阵中索引列向量，并将其作为列向量而不是1D数组返回。

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

给予

np.transpose(x[:,1])

不是解决方案。在numpy.transpose文档之后，这将返回行向量（1-D数组）。

Answer 1

几个选项 -

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