如果我想枚举一个数组(比如 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 的索引?
最佳答案
enumerate()
返回 (n, elem)
的序列对,其中 n
s 是连续的 Int
从零开始。这是有道理的,因为它是
SequenceType
的协议(protocol)扩展方法, 和一个任意序列
不一定具有与元素关联的索引。
你会得到你预期的结果
let powersSlice = slice.indices.map { pow(slice[$0], Double($0)) }
或
let powersSlice = zip(slice.indices, slice).map { pow($1, Double($0)) }
后一种方法可以推广到协议(protocol)扩展 任意集合的方法:
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))
}
}
关于swift - 如何使用原始索引枚举切片?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32688798/