如果我想枚举一个数组(比如对于一个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为基础的?
enumerate()
返回(n, elem)
对的序列,其中n
s是从0开始的连续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
用于"隐藏"特定的类型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))
}
}