J'ai une gamme de floats NUMPY 2D NUMPY, et je veux supprimer toutes les lignes dans lesquelles la troisième colonne de cette ligne contient une valeur inférieure à par exemple. x .
[[3,4,5], [3,3,8], [4,2,1], [1,2,1]] , avec seuil 2 , sorties [[3,4,5], [3,3,8]] .
3 Réponses :
Essayez celui-ci:
>>> import numpy as np
>>> x=np.array([[3,4,5],[3,3,8],[4,2,1],[1,2,1]])
>>> x=x[x[:,2]>=2]
>>> x
array([[3, 4, 5],
[3, 3, 8]])
upvoted..a une note latérale que cela fonctionne s'il y a une colonne 2 dans une colonne (si OP est intéressée): x [np.all (x> 2, axe = 1)]
Vous pouvez utiliser une liste-compréhension: sortie: Alternativement, vous pouvez utiliser filtre :
Essayez ceci:
import numpy as np array = np.array([[3,4,5],[3,3,8],[4,2,1],[1,2,1]]) array = np.array([x for x in array if x[2] > 2]) print (array)