在Swift中，是否有一种聪明的方法来使用数组上的高阶方法来返回5个第一个对象? obj-c的方法是保存一个索引，而for-loop则通过数组将索引值递增到5，然后返回新的数组。有办法做到这一点与过滤器，映射或减少?
当前回答
我稍微改变了Markus的答案，将其更新为最新的Swift版本，因为你的方法声明中的var不再被支持:
extension Array {
func takeElements(elementCount: Int) -> Array {
if (elementCount > count) {
return Array(self[0..<count])
}
return Array(self[0..<elementCount])
}
}
其他回答
我稍微改变了Markus的答案，将其更新为最新的Swift版本，因为你的方法声明中的var不再被支持:
extension Array {
func takeElements(elementCount: Int) -> Array {
if (elementCount > count) {
return Array(self[0..<count])
}
return Array(self[0..<elementCount])
}
}
你可以很容易地做到这一点，不需要filter, map, reduce或prefix，只需要通过下标返回数组的一个范围:
var wholeArray = [1, 2, 3, 4, 5, 6]
var n = 5
var firstFiveSlice = wholeArray[0..<n] // 1,2,3,4,5
let firstFiveArray = Array(firstFiveSlice)
Swift 4保存数组类型
extension Array {
func take(_ elementsCount: Int) -> [Element] {
let min = Swift.min(elementsCount, count)
return Array(self[0..<min])
}
}
简单明了
extension Array {
func first(elementCount: Int) -> Array {
let min = Swift.min(elementCount, count)
return Array(self[0..<min])
}
}
斯威夫特4
要获取Swift数组的前N个元素，可以使用prefix(_ maxLength: Int):
Array(largeArray.prefix(5))
