Utilisez numpy pour supprimer des colonnes ou des lignes contenant des éléments de certaines conditions

Utilisez numpy.where et numpy.delete en combinaison

>>> v = np.array(range(25)).reshape(5,5)
>>> v
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24]])
>>> np.delete(v, np.where(v>22)[1], 1)
array([[ 0,  1,  2],
       [ 5,  6,  7],
       [10, 11, 12],
       [15, 16, 17],
       [20, 21, 22]])
>>> np.delete(v, np.where(v>18)[0], 1)
array([[ 0,  1,  2],
       [ 5,  6,  7],
       [10, 11, 12],
       [15, 16, 17],
       [20, 21, 22]])
>>> np.delete(v, np.where(v>18)[0], 0)
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14]])
>>> 


Recommended Posts

Utilisez numpy pour supprimer des colonnes ou des lignes contenant des éléments de certaines conditions
Appliquez la fonction à la ligne ou à la colonne de numpy.array sans utiliser la notation d'inclusion de liste
Comment extraire les conditions (acquérir tous les éléments du Groupe qui remplissent les conditions) pour Groupe par Groupe
Convertir les éléments du tableau numpy de float en int
Comment utiliser numpy
Je veux déterminer l'authenticité d'un élément du tableau numpy
[Python] Comment utiliser l'instruction for. Une méthode d'extraction en spécifiant une plage ou des conditions.