在3D numpy数组中删除行

Question

我需要从3D numpy数组中删除一些行。对于exampe：

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]]])