如何使用原始索引枚举一个切片？

Question

如果我想枚举一个数组（例如用于 map() 函数，需要同时使用元素的索引和值），我可以使用 enumerate() 函数。例如：

import Foundation

let array: [Double] = [1, 2, 3, 4]

let powersArray = array.enumerate().map() {
    pow($0.element, Double($0.index))
}

print("array == \(array)")
print("powersArray == \(powersArray)")

// array == [1.0, 2.0, 3.0, 4.0]
// powersArray == [1.0, 2.0, 9.0, 64.0] <- As expected

现在，如果我想要使用数组中的某些子序列，我可以使用slice，它将允许我使用与原始数组相同的索引（这正是我在for循环中使用subscript访问器所需要的）。例如：

let range = 1..<(array.count - 1)
let slice = array[range]
var powersSlice = [Double]()

for index in slice.indices {
    powersSlice.append(pow(slice[index], Double(index)))
}

print("powersSlice == \(powersSlice)")

// powersSlice == [2.0, 9.0] <- As expected   

然而，如果我尝试像在原始数组中使用enumerate().map()方法一样，那么我会得到完全不同的行为。我将不再是通过slice的索引范围获取数据，而是得到一个新的基于0的索引范围：

let powersSliceEnumerate = slice.enumerate().map() {
    pow($0.element, Double($0.index))
}

print("powersSliceEnumerate == \(powersSliceEnumerate)")

// powersSliceEnumerate == [1.0, 3.0] <- Not as expected

问题是是否有一种好的方式（即不使用偏移量或其他手动调整）来使用切片自己的索引而不是自动生成的基于0的索引来枚举切片？

- 0x416e746f6e

---

不是针对你的问题的答案，但你可以用 let powersSlice = slice.indices.map { pow(slice[$0], Double($0)) } 替换 for 循环。 - Martin R

谢谢，@MartinR。我会记住你的建议，作为一个比仅仅偏移enumerate生成的索引范围更好的解决方法。 - 0x416e746f6e

Answer 1

enumerate() 返回一个由 (n, elem) 对组成的序列，其中 n 是从零开始的连续的整数。 这是有道理的，因为它是 SequenceType 的协议扩展方法，并且任意序列不一定与元素关联具有索引。

你可以通过以下方式获得你期望的结果

let powersSlice = slice.indices.map { pow(slice[$0], Double($0)) }

或者

let powersSlice = zip(slice.indices, slice).map { pow($1, Double($0)) }

后一种方法可以推广为任意集合的协议扩展方法：

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

这将返回一个由(index, elem)对组成的序列，其中index是集合中的索引，elem是相应的元素。使用AnySequence来“隐藏”从zip()返回的具体类型Zip2Sequence<RangeGenerator<Self.Index>, Self>，以免让调用者看到。

示例：

let powersSliceEnumerate = slice.indexEnumerate().map() { pow($0.element, Double($0.index)) }
print("powersSliceEnumerate == \(powersSliceEnumerate)")
// powersSliceEnumerate == [2.0, 9.0]

Swift 3 更新：

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

- Martin R

---

非常好。谢谢。 - 0x416e746f6e