Swiftで配列の最初の5つのオブジェクトを返す方法は？

Swiftで、配列の高次メソッドを使用して最初の5つのオブジェクトを返す賢い方法はありますか？それを行うobj-cの方法は、インデックスを保存し、インデックスが5になるまでインデックスをインクリメントしてループし、新しい配列を返すことでした。これを行う方法はありますかfilter、mapまたはreduce？

Swift配列の最初のN個の要素を取得する最も簡単な方法は、以下を使用することprefix(_ maxLength: Int)です。

let someArray = [1, 2, 3, 4, 5, 6, 7]
let first5 = someArray.prefix(5) // 1, 2, 3, 4, 5

これには、境界が安全であるという利点があります。渡したカウントがprefix配列のカウントよりも大きい場合は、配列全体が返されます。

注：コメントで指摘したように、Array.prefix実際に返しArraySlice、ありませんArray。ほとんどの場合、これで違いが生じることはありませんが、結果をArray型に割り当てる必要がある場合、またはArrayパラメーターを期待しているメソッドに渡す必要がある場合は、結果を型に強制する必要がありますArray。let first5 = Array(someArray.prefix(5))

更新：prefix配列の最初のn個の要素を取得するため に使用できるようになりました。接頭辞の使用方法の説明については、@ mluisbrownの回答を確認してください。

元の回答： なしでfilter、mapまたはreduce単に配列の範囲を返すだけで、非常に簡単に行うことができます。

var wholeArray = [1, 2, 3, 4, 5, 6]
var n = 5

var firstFive = wholeArray[0..<n] // 1,2,3,4,5

Swift 5では、必要に応じて、問題を解決するために次の6つのPlaygroundコードのいずれかを選択できます。

＃1。subscript(_:)下付き文字の使用

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。prefix(_:)メソッドの使用

複雑さ：コレクションがに準拠している場合は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"]

アップルの状態prefix(_:)：

最大長がコレクション内の要素数を超える場合、結果にはコレクション内のすべての要素が含まれます。

＃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"]

アップルの状態prefix(upTo:)：

このprefix(upTo:)メソッドを使用することは、コレクションの添え字として部分的に開いた部分的な範囲を使用することと同じです。添え字表記の方が優先されprefix(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"]

let a: [Int] = [0, 0, 1, 1, 2, 2, 3, 3, 4]
let b: [Int] = Array(a.prefix(5))
// result is [0, 0, 1, 1, 2]

SWIFT 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

配列の最初の5つの要素を取得するには、問題の配列をスライスするだけです。Swiftでは、次のようにします。array[0..<5]。

配列の最初のN個の要素の選択をもう少し機能的で一般化できるようにするには、それを行うための拡張メソッドを作成します。例えば：

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

varメソッド宣言内はサポートされなくなったため、Markusの回答を少し変更して最新のSwiftバージョンに更新しました。

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

スウィフト4

Swift配列の最初のN個の要素を取得するには、次を使用できますprefix(_ maxLength: Int)。

Array(largeArray.prefix(5))

配列タイプを保存するSwift 4

extension Array {
    func take(_ elementsCount: Int) -> [Element] {
        let min = Swift.min(elementsCount, count)
        return Array(self[0..<min])
    }
}

Swift 4の更新：

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

Swift 3の場合：

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

シンプルでシンプル

extension Array {
    func first(elementCount: Int) -> Array {
          let min = Swift.min(elementCount, count)
          return Array(self[0..<min])
    }
}

オブジェクトの配列の場合、Sequenceから拡張を作成できます。

extension Sequence {
    func limit(_ max: Int) -> [Element] {
        return self.enumerated()
            .filter { $0.offset < max }
            .map { $0.element }
    }
}

使用法：

struct Apple {}

let apples: [Apple] = [Apple(), Apple(), Apple()]
let limitTwoApples = apples.limit(2)

// limitTwoApples: [Apple(), Apple()]