Numpy splitting multidimensional arrays

Question

I have a multidimensional numpy array that I want to split based on a particular column.

Ex. [[1,0,2,3],[1,2,3,4],[2,3,4,5]] Say I want to split this array by the 2nd column with the expression x <=2. Then I would get two arrays [[1,0,2,3],[1,2,3,4]] and [[2,3,4,5]].

I am currently using this statement, which I don't think is correct.

splits = np.split(S, np.where(S[:, a] <= t)[0][:1]) #splits S based on t

#a is the column number

So when you say 2nd column, you mean 1-based. [1, 0, 2, 3], [1, 2, 3, 4] stay together because you compare 0 <= 2 and 2 <= 2. [2, 3, 4, 5] is "split" because you compare 3 <= 2. Correct? — orange
– orange, Commented Feb 19, 2016 at 6:31

orange · Accepted Answer · 2016-02-19 06:35:47Z

6

>>> import numpy as np
>>> a = np.asarray([[1,0,2,3],[1,2,3,4],[2,3,4,5]])
>>> a
array([[1, 0, 2, 3],
       [1, 2, 3, 4],
       [2, 3, 4, 5]])
>>> split1 = a[a[:,1] <= 2, :]
array([[1, 0, 2, 3],
       [1, 2, 3, 4]])
>>> split2 = a[a[:,1] > 2, :]
array([[2, 3, 4, 5]])

answered Feb 19, 2016 at 6:35

orange

8,25215 gold badges89 silver badges157 bronze badges

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

Collectives™ on Stack Overflow

Numpy splitting multidimensional arrays

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related