← retour aux snippets

numpy: where et masquage

Sélection et remplacements conditionnels.

python numpy #numpy#where#mask

objectif

Sélection et remplacements conditionnels.

code minimal

import numpy as np
x = np.array([1,-1,2])
print(np.where(x<0, 0, x).tolist())

utilisation

import numpy as np
x = np.array([1,2,3,4])
print(x[x%2==0].tolist())

variante(s) utile(s)

import numpy as np
x = np.array([1,2,3])
m = x>1; x[m] = 9
print(x.tolist())

notes

  • Masques booléens pour filtrage performant.