swift - 在 Swift 中使用索引遍历集合的正确且惯用的方法？

Question

我想遍历任意 Swift 集合并获取元素及其索引。

基本上可以替代:

for (idx, el) in collection.enumerate() {
    print("element at \(idx) is \(el)")
}

但这给了我真正的通用索引，而不仅仅是从 0 开始的连续整数。

当然，解决方案将成为接受任何类型集合的通用函数的一部分，否则差异不会很重要。

有没有比下面这样的简单循环更好的方法？

var idx = collection.startIndex, endIdx = collection.endIndex
while idx < endIdx {
    let el = collection[idx]
    print("element at \(idx) is \(el)")
    idx = idx.successor()
}

看起来很容易出错的写作。我知道我可以将该代码变成一个片段，但如果可能的话，我想找到一个更简洁、更惯用的解决方案。

Answer 1

对于任何集合，indices 属性返回有效范围指数。遍历索引和相应的元素同时，您可以使用 zip():

for (idx, el) in zip(collection.indices, collection) {
    print(idx, el)
}

数组切片示例:

let a = ["a", "b", "c", "d", "e", "f"]
let slice = a[2 ..< 5]

for (idx, el) in zip(slice.indices, slice) {
    print("element at \(idx) is \(el)")
}

输出:

element at 2 is celement at 3 is delement at 4 is e

You can define a custom extension method for that purpose(taken from How to enumerate a slice using the original indices?):

// Swift 2:
extension CollectionType {
    func indexEnumerate() -> AnySequence<(index: Index, element: Generator.Element)> {
        return AnySequence(zip(indices, self))
    }
}

// Swift 3:
extension Collection {
    func indexEnumerate() -> AnySequence<(Indices.Iterator.Element, Iterator.Element)> {
        return AnySequence(zip(indices, self))
    }
}

字符 View 示例:

let chars = "a😀🇩🇪z".characters
for (idx, el) in chars.indexEnumerate() {
    print("element at \(idx) is \(el)")
}

输出:

element at 0 is aelement at 1 is 😀element at 3 is 🇩🇪element at 7 is z