Elegant pythonic cumsum

前端 未结 10 1109
-上瘾入骨i
-上瘾入骨i 2020-12-16 11:54

What would be an elegant and pythonic way to implement cumsum?
Alternatively - if there\'a already a built-in way to do it, that would be even better of course...

10条回答
  •  时光取名叫无心
    2020-12-16 12:39

    a = [1, 2, 3 ,4, 5]
    
    # Using list comprehention
    cumsum = [sum(a[:i+1]) for i in range(len(a))]           # [1, 3, 6, 10, 15]
    
    # Using map()
    cumsum = map(lambda i:  sum(a[:i+1]), range(len(a)))     # [1, 3, 6, 10, 15]
    

提交回复
热议问题