Python listowego

Question

Powiel możliwe:
List comprehension for running total

Próbuję napisać zwięzłe listowych stworzyć CDF: Na przykład:

print f([0.2, 0.3,0.1,0.4]) 
[0.2,0.5,0.6,1.0] 

Standardowa procedura wyglądałaby tak (chcę napisać listę zrozumiałą dla e funkcja f()):

def f(probabilities) : 

    sum = 0 
    returnList = [] 
    for count in probabilities: 
     sum +=count 
     returnList = returnList + [sum] 
    return returnList 

Edytuj: Znalazłem funkcję numpy.cumsum(). Sprawdzę, czy korzysta ze spisanych słów.

Answer 1

Ta operacja jest tak powszechna, że ​​wiele języków (głównie funkcjonalnych, ale nie tylko) dostarcza do niej abstrakcje, zwykle o nazwie scanl (to jest jak reduce z wynikami pośrednimi). Nazwijmy go ireduce („iteracyjny zmniejszyć”):

def ireduce(f, state, it): 
    for x in it: 
     state = f(state, x) 
     yield state 

I teraz używać go:

import operator 

def f(probabilities): 
    return ireduce(operator.add, 0, probabilities) 

print(list(f([0.2, 0.3,0.1,0.4]))) 
# [0.2, 0.5, 0.6, 1.0] 

Answer 2

[sum(probabilities[:i+1]) for i in range(len(probabilities))] 

Ale nie rób tego, ponieważ jest to O (n^2). Zrozumienia listy w języku Python nie zostały zaprojektowane do tego celu. Użyj kodu proceduralnego, który już napisałeś.

Answer 3

To naprawdę nie jest ładna, i nie używa wyrażeń listowych, ale można to zrobić za pomocą zmniejszenia() funkcja, gdzie skumulowana wartość jest krotką posiadających aktualną sumę i listę wyników: brak

a = [0.2, 0.3, 0.1, 0.4] 
reduce((lambda result, val: (result[0] + val, result[1] + [result[0] + val])), a, (0, []))[1] 

Pythona z obsługą multi-line lambda czyni ten rodzaj brzydki. Używanie oddzielnej funkcji byłoby lepsze:

a = [0.2, 0.3, 0.1, 0.4] 
    def accumulate(result, val): 
     return (result[0] + val, result[1] + [result[0] + val]) 

    reduce(accumulate, a, (0, []))[1]