Add take, take_right
This commit is contained in:
18
snippets/take.md
Normal file
18
snippets/take.md
Normal file
@ -0,0 +1,18 @@
|
||||
---
|
||||
title: take
|
||||
tags: list,beginner
|
||||
---
|
||||
|
||||
Returns a list with `n` elements removed from the beginning.
|
||||
|
||||
Use slice notation to create a slice of the list with `n` elements taken from the beginning.
|
||||
|
||||
```py
|
||||
def take(itr, n = 1):
|
||||
return itr[:n]
|
||||
```
|
||||
|
||||
```py
|
||||
take([1, 2, 3], 5) # [1, 2, 3]
|
||||
take([1, 2, 3], 0) # []
|
||||
```
|
||||
18
snippets/take_right.md
Normal file
18
snippets/take_right.md
Normal file
@ -0,0 +1,18 @@
|
||||
---
|
||||
title: take_right
|
||||
tags: list,beginner
|
||||
---
|
||||
|
||||
Returns a list with `n` elements removed from the end.
|
||||
|
||||
Use slice notation to create a slice of the list with `n` elements taken from the end.
|
||||
|
||||
```py
|
||||
def take_right(itr, n = 1):
|
||||
return itr[-n:]
|
||||
```
|
||||
|
||||
```py
|
||||
take_right([1, 2, 3], 2) # [2, 3]
|
||||
take_right([1, 2, 3]) # [3]
|
||||
```
|
||||
Reference in New Issue
Block a user