ios - 按索引过滤数组

Question

我有一个元素数组。我还有一个 IndexSet，它指定需要将数组的哪些索引提取到新数组中。例如:

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

我想使用 swift filter 函数，但还没有得到答案。我该怎么做？

Answer 1

IndexSet 是递增整数的集合，因此您可以 将每个索引映射到相应的数组元素:

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

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

这假设所有索引对于给定的数组都是有效的。 如果不能保证，那么您可以过滤索引 (正如@dfri 正确评论的那样):

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