Listenverständnis für laufende Summe

Question

Ich möchte eine laufende Summe aus einer Liste von Zahlen erhalten.

Zu Demonstrationszwecken beginne ich mit einer fortlaufenden Liste von Zahlen mit range

a = range(20)

runningTotal = []
for n in range(len(a)):
    new = runningTotal[n-1] + a[n] if n > 0 else a[n]
    runningTotal.append(new)

# This one is a syntax error
# runningTotal = [a[n] for n in range(len(a)) if n == 0 else runningTotal[n-1] + a[n]]

for i in zip(a, runningTotal):
    print "{0:>3}{1:>5}".format(*i)

ergibt

  0    0
  1    1
  2    3
  3    6
  4   10
  5   15
  6   21
  7   28
  8   36
  9   45
 10   55
 11   66
 12   78
 13   91
 14  105
 15  120
 16  136
 17  153
 18  171
 19  190

Wie Sie sehen können, initialisiere ich eine leere Liste [] entonces append() in jeder Schleifeniteration. Gibt es einen eleganteren Weg zu diesem, wie eine Liste Verständnis?

Answer 1

Ein Listenverständnis hat keinen guten (sauberen, portablen) Weg, um auf die Liste selbst zu verweisen, die es erstellt. Ein guter und eleganter Ansatz könnte darin bestehen, die Aufgabe in einem Generator zu erledigen:

def running_sum(a):
  tot = 0
  for item in a:
    tot += item
    yield tot

um dies als Liste zu erhalten, verwenden Sie stattdessen natürlich list(running_sum(a)) .

Answer 2

Wenn Sie Folgendes verwenden können numpy hat es eine eingebaute Funktion namens cumsum das dies tut.

import numpy as np
tot = np.cumsum(a)  # returns a np.ndarray
tot = list(tot)     # if you prefer a list

Answer 3

Ich bin mir nicht sicher, ob das "elegant" ist, aber ich denke, dass die folgende Lösung viel einfacher und intuitiver ist (auf Kosten einer zusätzlichen Variablen):

a = range(20)

runningTotal = []

total = 0
for n in a:
  total += n
  runningTotal.append(total)

Die funktionelle Art, dasselbe zu tun, ist:

a = range(20)
runningTotal = reduce(lambda x, y: x+[x[-1]+y], a, [0])[1:]

...aber das ist viel weniger lesbar/pflegbar, etc.

@Omnifarous schlägt vor, dass dies verbessert werden sollte:

a = range(20)
runningTotal = reduce(lambda l, v: (l.append(l[-1] + v) or l), a, [0])

...aber ich finde das immer noch weniger unmittelbar nachvollziehbar als meinen ersten Vorschlag.

Erinnern Sie sich an die Worte von Kernighan: "Das Debuggen ist doppelt so schwer wie das Schreiben des Codes selbst. Wenn Sie also den Code so clever wie möglich schreiben, sind Sie per Definition nicht clever genug, um ihn zu debuggen."

Answer 4

Dies kann in 2 Zeilen in Python implementiert werden.

Durch die Verwendung eines Standardparameters entfällt die Notwendigkeit, eine Aux-Variable außerhalb zu pflegen, und wir führen lediglich eine map in die Liste aufzunehmen.

def accumulate(x, l=[0]): l[0] += x; return l[0];
map(accumulate, range(20))

Answer 5

Utilice itertools.accumulate() . Hier ist ein Beispiel:

from itertools import accumulate

a = range(20)
runningTotals = list(accumulate(a))

for i in zip(a, runningTotals):
    print "{0:>3}{1:>5}".format(*i)

Dies funktioniert nur mit Python 3. Unter Python 2 können Sie den Backport in der mehr-itertools Paket.