Comment puis-je calculer efficacement la somme cumulée d'une liste en Python ?

Somme cumulée en Python

Question :

Comment pouvons-nous trouver la somme cumulée des nombres dans une liste, comme [4, 4 6, 4 6 12]?

Introduction :

Le calcul de la somme cumulée implique d'ajouter à plusieurs reprises l'élément suivant à la somme cumulée. Bien que cela puisse être réalisé avec une boucle simple, il existe des approches plus efficaces utilisant NumPy ou des générateurs Python personnalisés.

Méthode NumPy :

La fonction cumsum de NumPy fournit une solution efficace :

import numpy as np

time_interval = [4, 6, 12]
np.cumsum(time_interval)
# Output: [4, 10, 22]

Cette approche est hautement optimisée et recommandée pour les opérations numériques sur de grandes tableaux.

Générateur Python :

Pour des solutions concises et économes en mémoire en Python pur, pensez à utiliser un générateur :

def accumu(iterable):
    yield next(iterable)
    for item in iterable:
        yield item + sum(accumu(iterable))

Utiliser ceci générateur :

list(accumu(time_interval))
# Output: [4, 10, 22]

Ce générateur accumule paresseusement des éléments, ce qui le rend approprié pour itérer sur de grandes séquences ou lorsque la mémoire est une préoccupation.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!