从NumPy数组中展开内部数组

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)

Answer 2

b = a.reshape((1,3,2))
print(b)
array([[[ 10,  10],
        [300, 300],
        [ 10, 300]]])