CODEWITHSUNDEEP
iosswift
Tometoyou
Tometoyou

Reputation: 8376

Filter array by indices

I have an array of elements. I also have an IndexSet that specifies which indices of the array need to be extracted into a new array. E.g.:

let array = ["sun", "moon", "star", "meteor"]
let indexSet: IndexSet = [2, 3]
// Some magic happens here to get:
let result = ["star", "meteor"]

I'm looking to use the swift filter function, but haven't got the answer yet. How can I do this?

Upvotes: 6

Views: 6650

Answers (3)

Benoît
Benoît

Reputation: 99

You can use Array.elements(at:)

let result = array.elements(at: indexSet)

Upvotes: -3

Martin R
Martin R

Reputation: 539685

IndexSet is a collection of increasing integers, therefore you can map each index to the corresponding array element:

let array = ["sun", "moon", "star", "meteor"]
let indexSet: IndexSet = [2, 3]

let result = indexSet.map { array[$0] } // Magic happening here!
print(result) // ["star", "meteor"]

This assumes that all indices are valid for the given array. If that is not guaranteed then you can filter the indices (as @dfri correctly remarked):

let result = indexSet.filteredIndexSet { $0 < array.count }.map { array[$0] }

Upvotes: 31

Luca Angeletti
Luca Angeletti

Reputation: 59496

You can use enumerated, filter and map like this

let result = array
    .enumerated()
    .filter { indexSet.contains($0.offset) }
    .map { $0.element }

Upvotes: 6

Related Questions