diff --git a/snippets/all_equal.md b/snippets/all_equal.md index bd6b0dff9..951561181 100644 --- a/snippets/all_equal.md +++ b/snippets/all_equal.md @@ -5,7 +5,7 @@ tags: list,beginner Checks if all elements in a list are equal. -- Use `set()` to eliminate duplicate elements and then use `len()` to check if length is 1. +- Use `set()` to eliminate duplicate elements and then use `len()` to check if length is `1`. ```py def all_equal(lst): diff --git a/snippets/index_of_all.md b/snippets/index_of_all.md new file mode 100644 index 000000000..846dcc684 --- /dev/null +++ b/snippets/index_of_all.md @@ -0,0 +1,18 @@ +--- +title: index_of_all +tags: list,intermediate +--- + +Returns a list of indices of all the occurrences of an element in a list. + +- Use `enumerate()` and a list comprehension to check each element for equality with `value` and adding `i` to the result. + +```py +def index_of_all(lst, value): + return [i for i, x in enumerate(lst) if x == value] +``` + +```py +index_of_all([1, 2, 1, 4, 5, 1], 1) # [0, 2, 5] +index_of_all([1, 2, 3, 4], 6) # [] +``` diff --git a/snippets/indices_of_occurence.md b/snippets/indices_of_occurence.md deleted file mode 100644 index f92220f5d..000000000 --- a/snippets/indices_of_occurence.md +++ /dev/null @@ -1,19 +0,0 @@ ---- -title: indices_of_occurrence -tags: list,beginner ---- - -Returns a list of indices of all the occurrences of an element in a list. - -- Use `enumerate()` to get index and value simultaneously while iterating. -- Use `==` to check equality, if current element `val` is equal to `value`. Then `idx` to list. - -```py -def indices_of_occurrence(lst, value): - return [idx for idx, val in enumerate(lst) if val == value] -``` - -```py -indices_of_occurrence([1, 2, 1, 4, 5, 1], 1) # [0, 2, 5] -indices_of_occurrence([1, 2, 3, 4], 6) # [] -```