← retour aux snippets

numpy: ufunc reduce et accumulate

Utiliser reduce/accumulate pour des scans rapides.

python numpy #numpy#ufunc#scan

objectif

Utiliser reduce/accumulate pour des scans rapides.

code minimal

import numpy as np

print(np.add.reduce([1,2,3,4]))

utilisation

import numpy as np

print(np.multiply.accumulate([1,2,3]).tolist())

variante(s) utile(s)

import numpy as np

x = np.array([1,3,2])
print(np.maximum.accumulate(x).tolist())

notes

  • accumulate renvoie tous les préfixes; reduce un scalaire.