Add cumsum
This commit is contained in:
20
snippets/cumsum.md
Normal file
20
snippets/cumsum.md
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
---
|
||||||
|
title: cumsum
|
||||||
|
tags: list,intermediate
|
||||||
|
---
|
||||||
|
|
||||||
|
Creates a list of partial sums.
|
||||||
|
|
||||||
|
- Use `itertools.accumulate()` to create the accumulated sum for each element.
|
||||||
|
- Use `list()` to convert the result into a list.
|
||||||
|
|
||||||
|
```py
|
||||||
|
from itertools import accumulate
|
||||||
|
|
||||||
|
def cumsum(lst):
|
||||||
|
return list(accumulate(lst))
|
||||||
|
```
|
||||||
|
|
||||||
|
```py
|
||||||
|
cumsum(range(0, 15, 3)) # [0, 3, 9, 18, 30]
|
||||||
|
```
|
||||||
Reference in New Issue
Block a user