删除3D numpy数组中的行

Question

我需要从3D numpy数组中删除一些行.例如:

a = [[1,2,3]
     [4,5,6]
     [7,8,9]

     [9,8,7]
     [6,5,4]
     [3,2,1]]

我想删除矩阵的两个页面的第三行,以获得类似的东西:

 a = [[1,2,3]
     [4,5,6]

     [9,8,7]
     [6,5,4]]

我试过了

   a = numpy.delete(a, 2, axis=0)

但我无法获得我所需要的东西.

Answer 1

axis应该1.

>>> import numpy
>>> a = [[[1,2,3],
...       [4,5,6],
...       [7,8,9]],
...      [[9,8,7],
...       [6,5,4],
...       [3,2,1]]]
>>> numpy.delete(a, 2, axis=1)
array([[[1, 2, 3],
        [4, 5, 6]],

       [[9, 8, 7],
        [6, 5, 4]]])