展开内部阵列

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)