在Swift中,是否有一种聪明的方法来使用数组上的高阶方法来返回5个第一个对象? obj-c的方法是保存一个索引,而for-loop则通过数组将索引值递增到5,然后返回新的数组。有办法做到这一点与过滤器,映射或减少?


当前回答

斯威夫特4

要获取Swift数组的前N个元素,可以使用prefix(_ maxLength: Int):

Array(largeArray.prefix(5))

其他回答

接这个分机。它修复了原来可怕的命名:

public extension Array {
    func first(_ count: Int) -> ArraySlice<Element> {
        return self.prefix(count)
    }
    
    func last(_ count: Int) -> ArraySlice<Element> {
        return self.suffix(count)
    }
}

用法:

someArr.first(5) 
someArr.last(5) 

我稍微改变了Markus的答案,将其更新为最新的Swift版本,因为你的方法声明中的var不再被支持:

extension Array {
    func takeElements(elementCount: Int) -> Array {
        if (elementCount > count) {
            return Array(self[0..<count])
        }
        return Array(self[0..<elementCount])
    }
}

使用Swift 5,根据您的需要,您可以选择以下6个Playground代码之一,以解决您的问题。


# 1。使用下标(_:)

let array = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
let arraySlice = array[..<5]
//let arraySlice = array[0..<5] // also works
//let arraySlice = array[0...4] // also works
//let arraySlice = array[...4] // also works
let newArray = Array(arraySlice)
print(newArray) // prints: ["A", "B", "C", "D", "E"]

# 2。使用前缀(_:)方法

复杂度:O(1)如果集合符合RandomAccessCollection;否则,O(k),其中k是要从集合开始选择的元素的数量。

let array = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
let arraySlice = array.prefix(5)
let newArray = Array(arraySlice)
print(newArray) // prints: ["A", "B", "C", "D", "E"]

Apple声明前缀(_:):

如果最大长度超过集合中的元素数量,则结果包含集合中的所有元素。


# 3。使用prefix(upTo:)方法

复杂性:O (1)

let array = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
let arraySlice = array.prefix(upTo: 5)
let newArray = Array(arraySlice)
print(newArray) // prints: ["A", "B", "C", "D", "E"]

Apple声明前缀(upTo:):

使用前缀(upTo:)方法相当于使用部分半开放范围作为集合的下标。下标表示法优先于前缀(upTo:)。


# 4。使用prefix(through:)方法

let array = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
let arraySlice = array.prefix(through: 4)
let newArray = Array(arraySlice)
print(newArray) // prints: ["A", "B", "C", "D", "E"]

# 5。使用removeSubrange(_:)方法

复杂度:O(n),其中n是集合的长度。

var array = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
array.removeSubrange(5...)
print(array) // prints: ["A", "B", "C", "D", "E"]

# 6。使用dropLast(_:)方法

复杂度:O(1)如果集合符合RandomAccessCollection;否则,O(k),其中k是要删除的元素的数量。

let array = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
let distance = array.distance(from: 5, to: array.endIndex)
let arraySlice = array.dropLast(distance)
let newArray = Array(arraySlice)
print(newArray) // prints: ["A", "B", "C", "D", "E"]

Prefix函数绝对是解决这个问题的最有效的方法,但是你也可以像下面这样使用for-in循环:

let array = [1,2,3,4,5,6,7,8,9]
let maxNum = 5
var iterationNumber = 0
var firstNumbers = [Int()]
if array.count > maxNum{
  for i in array{
    iterationNumber += 1
    if iterationNumber <= maxNum{
      firstNumbers.append(i)
    }
  }
  firstNumbers.remove(at: 0)
  print(firstNumbers)
} else {
  print("There were not \(maxNum) items in the array.")
}

这个解决方案占用了很多行代码,但会检查数组中是否有足够的项来执行程序,然后继续并解决问题。 该解决方案使用了许多基本函数,包括数组。Count,它返回数组中项的数量,而不是数组中最后一项的位置。它也使用数组。Append,它把东西添加到数组的末尾。最后,它使用数组。Remove,删除具有指定位置的数组项。

我已经测试过了,它至少适用于swift 5。

斯威夫特4

一个不同的解决方案:

一个简单的内联解决方案,不会崩溃,如果你的数组太短

[0,1,2,3,4,5].enumerated().compactMap{ $0.offset < 3 ? $0.element : nil }

但是用这个很好。

[0,1,2,3,4,5].enumerated().compactMap{ $0.offset < 1000 ? $0.element : nil }

如果你这样做,通常会崩溃:

[0,1,2,3,4,5].prefix(upTo: 1000) // THIS CRASHES

[0,1,2,3,4,5].prefix(1000) // THIS DOESNT