展開內部陣列

Question

我怎樣才能改變這種numpy陣列：

[[[10 10]] 
[[300 300]] 
[[10 300]]] 

進入這一個：

[[[ 10 10] 
    [300 300] 
    [ 10 300]]] 

Answer 1

您可以使用advanced indexing切子陣的第一個項目，然後將其包裹在外部陣列中：

a = numpy.array([[[10, 10]], 
    [[300, 300]], 
    [[10, 300]]]) 
b = numpy.array([a[:,0]]) 
print(b) 

打印

[[[ 10 10] 
    [300 300] 
    [ 10 300]]] 

或者，使用swapaxes：

b = numpy.swapaxes(a, 1, 0)