Delete some array elements from numpy array

Question

I have the numpy array:

a = np.array([[ 255,255,255],
              [ 255,2,255],
              [ 3,123,23],
              [ 255,255,255],
              [ 0, 255, 3]])

And I want to delete all the elements with [ 255,255,255], the result should be:

[[ 255,2,255],
 [ 3,123,23],
 [ 0, 255, 3]])

I tried with:

import numpy as np
a = np.array([[ 255,255,255],
              [ 255,2,255],
              [ 3,123,23],
              [ 255,255,255],
              [ 0, 255, 3]])

np.delete(a, [255,255,255])

but nothing happens.

CDJB · Accepted Answer · 2019-11-14 13:12:37Z

4

You can do this:

np.array([x for x in a if np.any(x != 255)])

which gives:

array([[255,   2, 255],
       [  3, 123,  23],
       [  0, 255,   3]])

Edit: To avoid list comprehensions -

np.delete(a, np.where((a == 255).all(axis=1)), axis=0)

edited Nov 14, 2019 at 13:12

answered Nov 14, 2019 at 13:05

CDJB

14.6k6 gold badges36 silver badges58 bronze badges

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

Comments

Yeghishe Kerobyan · Accepted Answer · 2019-11-14 13:15:14Z

3

here is a fast vectorized way to do it

a[(a!=255).any(axis=1),:]
Out[136]: 
array([[255,   2, 255],
       [  3, 123,  23],
       [  0, 255,   3]])

answered Nov 14, 2019 at 13:15

Yeghishe Kerobyan

1497 bronze badges

Collectives™ on Stack Overflow

Delete some array elements from 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