如何从 Scala 数组中选择随机元素？

Question

例如，有一个 Scala 数组 val A = Array("please", "help", "me")。如何从这个数组中选择一个随机元素？

Answer 1

如果您想要更惯用的解决方案，请考虑使用 typeclass 模式（scala 中的隐式类）。

implicit class ListOps[A](list: List[A]) {
  def getRandomElement: Option[A] = list match {
    case Nil => None
    case _ => list.lift(scala.util.Random.nextInt(list.size))
  }
  def randomChoice(n: Int): Option[List[A]] =
    (1 to n).toList.foldLeft(Option(List[A]()))((acc, e) => getRandomElement.flatMap(r => acc.map(a => a :+ r)))
}

现在，如果隐式类在范围内，您可以：

val randomElement: Option[String] = List("this", "is", "a", "list").getRandomElement

如果您确定该选项包含某些值，则可以使用 get 方法。

randomElement.get // This will return a String (or a NotSuchElementExeption)

尽管如此，还是建议使用模式匹配或 getOrElse：

randomElement match {
  case None => ??? // This is what you do when a None is encounter (e.g. for empty lists)
  case Some(result) => ??? // The variable result contains a string. 

注意，randomChoice 方法假定元素替换。

Answer 2

根本不涉及重新排列数组的更好答案是这样的：

import scala.util.Random

object sample {
  //gets random element from array
  def arr[T](items:Array[T]):T = {
    items(Random.nextInt(items.length))
  }
}

这也通常有效

Answer 3

import scala.util.Random

val A = Array("please", "help", "me")
Random.shuffle(A.toList).head

Answer 4

import scala.util.Random

val A = List(1, 2, 3, 4, 5, 6)
A(Random.nextInt(A.size))

Answer 5

import java.util.Random
// ...
val rand = new Random(System.currentTimeMillis())
val random_index = rand.nextInt(A.length)
val result = A(random_index)

Answer 6

我们还可以使用 Option monad 添加一些安全性（使用 lift 方法）。

实际上，在任何集合上使用此方法时，
即使您的集合为空，或者您的随机索引超出范围，您的结果也将始终是一个选项。

安全驾驶<3

def getRandElemO[T](arr: Array[T]): Option[T] =
  if (arr.isEmpty) None
  else arr.lift(util.Random.nextInt(arr.length))

// or the one liner:
// def getRandElemO[T](arr: Array[T]): Option[T] =
//   arr.headOption.flatMap(_ => arr.lift(util.Random.nextInt(arr.length)))