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

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

现在,如果我想使用数组中的一些子序列,我可以使用a slice,它将允许我使用与我在原始数组中使用的相同的索引(这正是我想要的情况,如果我会subscript在for循环中使用访问器).例如:

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的切片？

Answer 1

enumerate()返回(n, elem)一对对的序列,其中ns是Int从零开始的连续s.这是有道理的,因为它是一种协议扩展方法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))
    }
}