Vectorisation du calcul d'un histogramme

J’étais en train de coder un truc qui utilise une sorte d’histogramme et je me suis demandé si on pouvais s’affranchir des boucles en utilisant numpy (ou autre). En entrée j’ai les bins et j’aimerais dire à quelle classe appartient chaque exemple. Un exemple :

data = [1, 18, 21, 0, 48, 21, 44]
bins = [5, 20, 30, 50]

out = [0, 1, 2, 0, 3, 2, 3]

J’ai donc réussi à vectoriser cette opération, je me demandais si vous aviez d’autres solutions ou ce que vous pensiez de la mienne.

    data = ...
    bins = ...
    data = np.repeat(data[..., np.newaxis], len(bins), 0)
    out = np.argmax(data < bins, 1)

Merci

Salut @Ricocotam,

Il n’y a pas besoin d’utiliser repeat. Numpy propage les opérations sur certaines dimensions.

out = np.argmax(data[..., np.newaxis] < bins, axis=1)