垂直打印 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選項）