← retour aux snippets

numpy: fancy indexing

Indexer avec listes/arrays d'indices et masques booléens.

objectif

Indexer avec listes/arrays d’indices et masques booléens.

code minimal

import numpy as np
x = np.array([10,20,30,40])
print(x[[0,2]].tolist())

utilisation

import numpy as np
x = np.arange(6).reshape(3,2)
print(x[[0,2], [1,0]].tolist())

variante(s) utile(s)

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

notes

  • Copies créées; utiliser vues si possible pour performance.