如何使用原始索引枚举切片？答案

【问题标题】：How to enumerate a slice using the original indices?如何使用原始索引枚举切片？
【发布时间】：2015-12-17 18:34:38
【问题描述】：

如果我想枚举一个数组（比如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() 方法，然后我会得到完全不同的行为。我会得到一个新的、基于 0 的范围，而不是 slice 的索引范围：

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

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

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

问题是是否有一种体面的方式（即无需使用偏移量进行手动调整等）来使用自己的索引而不是自动生成的基于 0 的索引来枚举切片？

【问题讨论】：

不是您问题的答案，但您可以将 for-loop 替换为 let powersSlice = slice.indices.map { pow(slice[$0], Double($0)) }
谢谢，@MartinR。我会记住您的建议，作为一种解决方法，而不是仅仅抵消 enumerate 生成的索引范围。

标签： swift slice enumerate

【解决方案1】：

enumerate() 返回(n, elem) 对的序列，其中ns 是从零开始的连续Ints。这是有道理的，因为它是 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 用于“隐藏”特定类型 Zip2Sequence<RangeGenerator<Self.Index>, Self>从zip()返回来自调用者。

例子：

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))
    }
}

【讨论】：

优秀。谢谢。