select required data from numpy array

Question

Want to select the data between 1 and 4, and convert others as np.nan But what's is the soluton?

import numpy as np
data = np.array([1,2,3,4,5])
selected = np.where(1<data<4, data, np.nan)
print (selected)

Traceback (most recent call last):
  File "C:/Users/fe/Desktop/t.py", line 3, in <module>
    selected = np.where(1<data<4, data, np.nan)
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

mdml · Accepted Answer · 2013-12-18 15:15:53Z

4

You are very close, you just need a different way to select the relevant indices in data. Try:

>>> selected = np.where((data < 4) & (data > 1), data, np.nan)
>>> selected
array([ nan,   2.,   3.,  nan,  nan])

(data < 4) & (data > 1) finds the indices of the data that are BOTH < 4 and >1.

answered Dec 18, 2013 at 15:15

mdml

23k8 gold badges61 silver badges66 bronze badges

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

Collectives™ on Stack Overflow

select required data from numpy array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related