python - 从格式化为列/行的python-numpy矩阵获取列/行向量

Question

当我在 python-numpy 中选择一列或一行时，我总是必须重新格式化它，以便它成为一个实际的列或行向量。所以我总是必须像我的示例末尾那样编写代码，这在我看来完全是过度设计的。有没有办法从矩阵中选择列或行，以直接获得正确的格式化向量？

>>> import numpy as np
>>> A = np.zeros((2,3))
>>> # Show the size of the matrix
>>> print(A.shape)
(2, 3)
>>> # This is not a row vector
>>> print(A[1,:].shape)
(3,)
>>> # Get a real row vector (looks totally over-engineered)
>>> print(np.array([A[1,:]]).shape)
(1, 3)

标签： pythonnumpynumpy-ndarray