Tuple to index using numpy

Question

It drives me crazy, but I can't figure it out I have a data matrix of (10000,4)

I need to select some rows where the elements of column 0

ind1=np.where( (data[:,0]>55) & (data[:,0]<65) )

I want to keep that data only so

keep_data=data[ind1,:]

But keep_data is now (1,10000,4)

Why is that?

P.S. What i do is th efollwing

keep_data=np.reshape(keep_data,(keep_data.shape[1],keep_data.shape[2]))

jpp · Accepted Answer · 2018-05-24 13:20:02Z

2

numpy.where returns a tuple.

Therefore, use ind1 = np.where((data[:,0]>55) & (data[:,0]<65))[0]

Notice the [0] indexing to select the only element of the tuple.

This is noted in the docs:

numpy.where(condition[, x, y])

Return elements, either from x or y, depending on condition.

If only condition is given, return the tuple condition.nonzero(), the indices where condition is True.

edited May 24, 2018 at 13:20

answered May 24, 2018 at 13:17

jpp

166k37 gold badges301 silver badges362 bronze badges

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

Collectives™ on Stack Overflow

Tuple to index using numpy

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