Tableaux multidimensionnels fractionnés numpy

Question

J'ai un tableau numpy multidimensionnel que je veux diviser en fonction d'une colonne particulière.

Ex. [[1,0,2,3],[1,2,3,4],[2,3,4,5]] Dites que je veux diviser ce tableau par la 2ème colonne avec l'expression x <=2. Ensuite, je voudrais obtenir deux tableaux [[1,0,2,3],[1,2,3,4]] et . J'utilise actuellement cette déclaration, que je ne pense pas correcte.

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

#a is the column number 

Answer 1

>>> 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]])