Select elements in numpy

Asked Jul 10 '17 at 18:28

Active Jul 10 '17 at 18:29

Viewed 78 times

I have a numpy array with random values like this:

In [493]: arr = np.random.randint(-10,10,(5,7))
In [494]: arr
Out[494]: 
array([[ -5,  -6,  -7,  -1,  -8,  -8, -10],
       [ -9,   1,  -3,  -9,   3,   8,  -1],
       [  6,  -7,   4,   0,  -4,   4,  -2],
       [ -3, -10,  -2,   7,  -4,   2,   2],
       [ -5,   5,  -1,  -7,   7,   5,  -7]])

Now, I want to do a filter method:

if (elem > -3 && elme < 3), elem <- 0

How to implement this method in numpy without using for loop?

edited Jul 10 '17 at 18:29

ayhan

64,199
17
170
189

asked Jul 10 '17 at 18:28

dodolong

1

In questions with random data, it can be helpful to set a value for `np.random.seed(value)` so that your problem can be reproducible as-is. – Brad Solomon Jul 10 '17 at 18:31
3

`arr[(arr > -3) & (arr < 3)] = 0` . Look for boolean indexing. (you can start from [here](https://docs.scipy.org/doc/numpy/user/basics.indexing.html)). – ayhan Jul 10 '17 at 18:32
What's `elem – cs95 Jul 10 '17 at 18:34
@cᴏʟᴅsᴘᴇᴇᴅ Set `elem` to zero – Moses Koledoye Jul 10 '17 at 18:35

Select elements in numpy

0 Answers0