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