Python lista por comprensión

Question

Duplicar posible:
List comprehension for running total

Estoy intentando escribir una declaración lista por comprensión concisa para crear una función de distribución: Por ejemplo:

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

Un procedimiento estándar se vería así (quiero escribir una lista de comprensión para e función f()):

def f(probabilities) : 

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

Editar: Encontré una función numpy.cumsum(). Verifico si usa listas de comprensión.

Answer 1

Esa operación es tan común que muchos idiomas (principalmente los funcionales, pero no solo) proporcionan abstracciones para ello, generalmente con el nombre scanl (es como un reduce con resultados intermedios). Digamos que es ireduce ("iterativo reducir"):

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

Y ahora utilizarlo:

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))] 

Pero no hagas eso porque es O (n^2). Las comprensiones de la lista Python no fueron diseñadas para esto. Usa el código de procedimiento que ya escribiste.

Answer 3

No es muy bonita, y no está usando listas por comprensión, pero se puede hacer esto con el reducir falta de apoyo a la lambda de varias líneas

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

de Python hace que este tipo de feo: (función, cuando el valor acumulado es una tupla que sostiene la suma actual y la lista de resultados). Usar una función separada sería mejor:

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]