diff --git a/snippets/take.md b/snippets/take.md new file mode 100644 index 000000000..6b1f77b7b --- /dev/null +++ b/snippets/take.md @@ -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) # [] +``` diff --git a/snippets/take_right.md b/snippets/take_right.md new file mode 100644 index 000000000..0e68fcbb8 --- /dev/null +++ b/snippets/take_right.md @@ -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] +```