Python的numpy的數組操作

Question

我需要以某種方式使該數組：

[[639 190] 
[ 44 1] 
[ 71 4] 
..., 
[863 347] 
[870 362] 
[831 359]] 

這個樣子的：

[[[639 190]] 
[[ 44 1]] 
[[ 71 4]] 
..., 
[[863 347]] 
[[870 362]] 
[[831 359]]] 

我會怎麼做呢？我是新來的，我需要它來進行科學實驗。

Answer 1

添加一個新的軸與None/np.newaxis -

a[:,None,:] # Or simply a[:,None] 

採樣運行 -

In [222]: a = np.random.randint(0,9,(4,3)) 

In [223]: a 
Out[223]: 
array([[1, 6, 6], 
     [4, 4, 5], 
     [7, 4, 4], 
     [4, 1, 3]]) 

In [224]: a[:,None] 
Out[224]: 
array([[[1, 6, 6]], 

     [[4, 4, 5]], 

     [[7, 4, 4]], 

     [[4, 1, 3]]]) 

Answer 2

除了通過@Divakar提到newaxis/None，

np.expand_dims(input_array, axis=1)