ios - How to get index(of:) to return multiple indices?

Question

Here is a group array.

var group = ["H","H","E","D",
             "G","D","G","E",
             "D","B","A","B",
             "A","A","G","C",
             "C","H","D","G",
             "H","B","E","F",
             "F","C","E","A",
             "B","C","F","F"]

I want to do something like this to find indices of "A".

group.index(of: "A"!)

But this will return only first index, but not other indices for next three "A"s.

print(group.index(of: "A")!) //10

What do I do to get the program to return all four indices for "A"?

Answer 1

You might use a combination of enumerated and compactMap:

let indexArray = group.enumerated().compactMap {
   $0.element == "A" ? $0.offset : nil
}    
print(indexArray) // [10, 12, 13, 27]