Comment remplacer les valeurs négatives par zéro dans NumPy

Vous pouvez utiliser la syntaxe de base suivante pour remplacer les valeurs négatives par zéro dans NumPy :

my_array[my_array < 0] = 0

Cette syntaxe fonctionne avec les tableaux NumPy 1D et 2D.

Les exemples suivants montrent comment utiliser cette syntaxe dans la pratique.

Exemple 1 : Remplacer les valeurs négatives par zéro dans un tableau NumPy 1D

Le code suivant montre comment remplacer toutes les valeurs négatives par zéro dans un tableau NumPy :

import numpy as np

#create 1D NumPy array
my_array = np.array([4, -1, 6, -3, 10, 11, -14, 19, 0])

#replace negative values with zero in array
my_array[my_array < 0] = 0

#view updated array
print(my_array)

[ 4  0  6  0 10 11  0 19  0]

Notez que chaque valeur négative du tableau d’origine a été remplacée par zéro.

Exemple 2 : Remplacer les valeurs négatives par zéro dans un tableau NumPy 2D

Supposons que nous ayons le tableau NumPy 2D suivant :

import numpy as np

#create 2D NumPy array
my_array = np.array([3, -5, 6, 7, -1, 0, -5, 9, 4, 3, -5, 1]).reshape(4,3)

#view 2D NumPy array
print(my_array)

[[ 3 -5  6]
 [ 7 -1  0]
 [-5  9  4]
 [ 3 -5  1]]

Nous pouvons utiliser le code suivant pour remplacer toutes les valeurs négatives par zéro dans le tableau NumPy :

#replace all negative values with zero in 2D array
my_array[my_array < 0] = 0

#view updated array
print(my_array)

[[3 0 6]
 [7 0 0]
 [0 9 4]
 [3 0 1]]

Notez que toutes les valeurs négatives du tableau 2D d’origine ont été remplacées par zéro.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes dans NumPy :

Comment remplir un tableau NumPy avec des valeurs
Comment supprimer des éléments spécifiques du tableau NumPy
Comment remplacer des éléments dans un tableau NumPy
Comment obtenir une ligne spécifique à partir d’un tableau NumPy