diff --git a/snippets/indices_of_occurence.md b/snippets/indices_of_occurence.md new file mode 100644 index 000000000..f92220f5d --- /dev/null +++ b/snippets/indices_of_occurence.md @@ -0,0 +1,19 @@ +--- +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) # [] +```