Remove rows in 3D numpy array

Question

I need to remove some rows from a 3D numpy array. For exampe:

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

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

and I want to remove the third row of both pages of the matrix in order to obtain something like:

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

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

I have tried with

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

but I can't obtain what I need.

falsetru · Accepted Answer · 2013-10-24 09:24:18Z

10

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

answered Oct 24, 2013 at 9:24

falsetru

371k69 gold badges769 silver badges659 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Mike Altonji · Accepted Answer · 2020-12-04 02:11:32Z

0

A different approach would be to use slice notation, where you remove the last row of the 2nd axis.

>>> a[:, :-1, :]
array([[[1, 2, 3],
        [4, 5, 6]],

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

answered Dec 4, 2020 at 2:11

Mike Altonji

4053 silver badges12 bronze badges

Collectives™ on Stack Overflow

Remove rows in 3D numpy array

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related