Получить случайные элементы из массива в swift

у меня есть массив типа:

var names: String = [ "Peter", "Steve", "Max", "Sandra", "Roman", "Julia" ]

Я хотел бы получить 3 случайных элементов из массива. Я иду с C#, но в swift я не уверен, с чего начать. Я думаю, что я должен сначала перетасовать массив, а затем выбрать первые 3 элемента из него, например?

Я попытался перетасовать его со следующим расширением:

extension Array
{
    mutating func shuffle()
    {
        for _ in 0..<10
        {
            sort { (_,_) in arc4random() < arc4random() }
        }
    }
}

но затем он говорит, что" '() ' не конвертируется в "[Int]" "в месте" shuffle ()".

для выбора ряда элементы, которые я использую:

var randomPicks = names[0..<4];

который выглядит хорошо до сих пор.

как тасовать? Или у кого есть лучше/более элегантное решение для этого?

11
arrays random shuffle swift

4 ответов

Xcode 9 * Swift 4

extension Array {
    /// Returns an array containing this sequence shuffled
    var shuffled: Array {
        var elements = self
        return elements.shuffle()
    }
    /// Shuffles this sequence in place
    @discardableResult
    mutating func shuffle() -> Array {
        let count = self.count
        indices.lazy.dropLast().forEach {
            swapAt(, Int(arc4random_uniform(UInt32(count - ))) + )
        }
        return self
    }
    var chooseOne: Element { return self[Int(arc4random_uniform(UInt32(count)))] }
    func choose(_ n: Int) -> Array { return Array(shuffled.prefix(n)) }
}

тестирование игровой площадки

var alphabet = ["A","B","C","D","E","F","G","H","I","J","K","L","M","N","O","P","Q","R","S","T","U","V","W","X","Y","Z"]
let shuffledAlphabet = alphabet.shuffled

let letter = alphabet.chooseOne

var numbers = Array(0...9)

let shuffledNumbers = numbers.shuffled
shuffledNumbers                              // [8, 9, 3, 6, 0, 1, 4, 2, 5, 7]

numbers            // [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

numbers.shuffle() // mutate it  [6, 0, 2, 3, 9, 1, 5, 7, 4, 8]

numbers            // [6, 0, 2, 3, 9, 1, 5, 7, 4, 8]

let pick3numbers = numbers.choose(3)  // [8, 9, 2]

Xcode 8.3.1 * Swift 3.1

import UIKit

extension Array {
    /// Returns an array containing this sequence shuffled
    var shuffled: Array {
        var elements = self
        return elements.shuffle()
    }
    /// Shuffles this sequence in place
    @discardableResult
    mutating func shuffle() -> Array {
        let count = self.count
        indices.lazy.dropLast().forEach {
            guard case let index = Int(arc4random_uniform(UInt32(count - ))) + , index !=  else { return }
            swap(&self[], &self[index])
        }
        return self
    }
    var chooseOne: Element { return self[Int(arc4random_uniform(UInt32(count)))] }
    func choose(_ n: Int) -> Array { return Array(shuffled.prefix(n)) }
}
25

или у кого-нибудь есть лучше/более элегантное решение для этого?

Я делаю. Алгоритмически лучше, чем принятый ответ, который действительно считается-1 arc4random_uniform операции для полной перетасовки, мы можем просто выбрать n значения n arc4random_uniform операции.

и на самом деле, я получил два способы сделать лучше, чем принятый ответ:

лучшим решением

extension Array {
    /// Picks `n` random elements (straightforward approach)
    subscript (randomPick n: Int) -> [Element] {
        var indices = [Int](0..<count)
        var randoms = [Int]()
        for _ in 0..<n {
            randoms.append(indices.remove(at: Int(arc4random_uniform(UInt32(indices.count)))))
        }
        return randoms.map { self[] }
    }
}

лучший решение

следующее решение в два раза быстрее, чем предыдущий.

для Swift 3.0 и 3.1

extension Array {
    /// Picks `n` random elements (partial Fisher-Yates shuffle approach)
    subscript (randomPick n: Int) -> [Element] {
        var copy = self
        for i in stride(from: count - 1, to: count - n - 1, by: -1) {
            let j = Int(arc4random_uniform(UInt32(i + 1)))
            if j != i {
                swap(&copy[i], &copy[j])
            }
        }
        return Array(copy.suffix(n))
    }
}

для Swift 3.2 и 4.x

extension Array {
    /// Picks `n` random elements (partial Fisher-Yates shuffle approach)
    subscript (randomPick n: Int) -> [Element] {
        var copy = self
        for i in stride(from: count - 1, to: count - n - 1, by: -1) {
            copy.swapAt(i, Int(arc4random_uniform(UInt32(i + 1))))
        }
        return Array(copy.suffix(n))
    }
}

использование:

let digits = Array(0...9)  // [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
let pick3digits = digits[randomPick: 3]  // [8, 9, 0]
7

Swift 4.1 и ниже 

let playlist = ["Nothing Else Matters", "Stairway to Heaven", "I Want to Break Free", "Yesterday"]
let index = Int(arc4random_uniform(UInt32(playlist.count)))
let song = playlist[index]

Swift 4.2 и выше

if let song = playlist.randomElement() {
  print(song)
} else {
  print("Empty playlist.")
}
2

вы также можете использовать arc4random() выбрать три элемента из массива. Что-то вроде этого:--2--> 

extension Array {
    func getRandomElements() -> (T, T, T) {
        return (self[Int(arc4random()) % Int(count)],
                self[Int(arc4random()) % Int(count)],
                self[Int(arc4random()) % Int(count)])
    }
}

let names = ["Peter", "Steve", "Max", "Sandra", "Roman", "Julia"]
names.getRandomElements()

Это просто пример, вы также можете включить логику в функцию, чтобы получить другое имя для каждого из них.

1