垂直打印 numpy 数组的一维切片

Question

我正在尝试打印 numpy 数组的垂直切片，以便它垂直显示但始终水平打印。 给定这个方形数组：

a = np.ones([5,5])

我试过了：

print a[:,1]
print np.reshape(a[:,1], (1,-1))
print a[:,1].T
print [a[:,1].T]

这给我：

[ 1.  1.  1.  1.  1.]
[[ 1.  1.  1.  1.  1.]]
[ 1.  1.  1.  1.  1.]
[array([ 1.,  1.,  1.,  1.,  1.])]

我想看看：

[[1],
 [1],
 [1],
 [1], 
 [1]]

Answer 1

您需要添加一个新轴：

a[:, 1, None]
Out: 
array([[ 1.],
       [ 1.],
       [ 1.],
       [ 1.],
       [ 1.]])

要么

a[:, 1, np.newaxis]
Out: 
array([[ 1.],
       [ 1.],
       [ 1.],
       [ 1.],
       [ 1.]])

Answer 2

我将第二个索引器放在方括号中

a[:, [1]]

array([[ 1.],
       [ 1.],
       [ 1.],
       [ 1.],
       [ 1.]])

Answer 3

添加尺寸的另一种方法：

a[:,1:2]
Out:
array([[ 1.],
   [ 1.],
   [ 1.],
   [ 1.],
   [ 1.]])

Answer 4

你也可以使用np.vstack() ：

print(np.vstack(a[:,1]))
[[1.]
 [1.]
 [1.]
 [1.]
 [1.]]

Answer 5

另一个选择是，我有时使用atleast_2d ：

np.atleast_2d(a[:, 1]).T

（也有atleast_1d和atleast_3d选项）