← retour aux snippets

numpy: unique avec counts

Valeurs distinctes et effectifs rapidement.

python numpy #numpy#unique#counts

objectif

Valeurs distinctes et effectifs rapidement.

code minimal

import numpy as np
u, c = np.unique([1,1,2,3], return_counts=True)
print(dict(zip(u.tolist(), c.tolist()))[1])

utilisation

import numpy as np
u, idx = np.unique([2,1,2,3], return_index=True)
print(u[idx.argsort()].tolist() is not None)

variante(s) utile(s)

import numpy as np
u, inv = np.unique([1,1,2], return_inverse=True)
print(inv.tolist())

notes

  • return_inverse permet de remapper vers les indices uniques.