web-dev-qa-db-fra.com

Min/max avec Option [T] pour une Seq éventuellement vide?

Je fais un peu de gymnastique Scala où j'ai Seq[T] dans lequel j'essaie de trouver l'élément "le plus petit". C'est ce que je fais en ce moment:

val leastOrNone = seq.reduceOption { (best, current) =>
    if (current.something < best.something) current
    else best
}

Cela fonctionne bien, mais je ne suis pas tout à fait satisfait - c'est un peu long pour une chose aussi simple, et je me fiche bien de "if" s . Utiliser minBy serait beaucoup plus élégant:

val least = seq.minBy(_.something)

... mais min et minBy lèvent des exceptions lorsque la séquence est vide. Existe-t-il un moyen idiomatique et plus élégant de trouver le plus petit élément d’une liste éventuellement vide en tant que Option?

scalascala-collections
47
gustafc

Démarrer Scala 2.13, minByOption / maxByOption fait maintenant partie de la bibliothèque standard et renvoie None si la séquence est vide:

seq.minByOption(_.something)

List((3, 'a'), (1, 'b'), (5, 'c')).minByOption(_._1) // Option[(Int, Char)] = Some((1,b))
List[(Int, Char)]().minByOption(_._1)                // Option[(Int, Char)] = None
9
Xavier Guihot
seq.reduceOption(_ min _)

fait ce que tu veux?

Edit: Voici un exemple incorporant votre _.something:

case class Foo(a: Int, b: Int)
val seq = Seq(Foo(1,1),Foo(2,0),Foo(0,3))
val ord = Ordering.by((_: Foo).b)
seq.reduceOption(ord.min)  //Option[Foo] = Some(Foo(2,0))

ou, comme méthode générique:

def minOptionBy[A, B: Ordering](seq: Seq[A])(f: A => B) = 
  seq reduceOption Ordering.by(f).min

que vous pouvez invoquer avec minOptionBy(seq)(_.something)

66
Luigi Plinge

Une version sûre, compacte et O(n) avec Scalaz:

xs.nonEmpty option xs.minBy(_.foo)
8
Erik Allik

Pas vraiment une option pour une liste plus longue en raison de la complexité de O(nlogn):

seq.sortBy(_.something).headOption
4
Tomasz Nurkiewicz

Que dis-tu de ça?

import util.control.Exception._
allCatch opt seq.minBy(_.something)

Ou, plus verbeux, si vous ne voulez pas avaler d’autres exceptions:

catching(classOf[UnsupportedOperationException]) opt seq.minBy(_.something)

Alternativement, vous pouvez pimp toutes les collections avec quelque chose comme ça:

import collection._

class TraversableOnceExt[CC, A](coll: CC, asTraversable: CC => TraversableOnce[A]) {

  def minOption(implicit cmp: Ordering[A]): Option[A] = {
    val trav = asTraversable(coll)
    if (trav.isEmpty) None
    else Some(trav.min)
  }

  def minOptionBy[B](f: A => B)(implicit cmp: Ordering[B]): Option[A] = {
    val trav = asTraversable(coll)
    if (trav.isEmpty) None
    else Some(trav.minBy(f))
  }
}

implicit def extendTraversable[A, C[A] <: TraversableOnce[A]](coll: C[A]): TraversableOnceExt[C[A], A] =
  new TraversableOnceExt[C[A], A](coll, identity)

implicit def extendStringTraversable(string: String): TraversableOnceExt[String, Char] =
  new TraversableOnceExt[String, Char](string, implicitly)

implicit def extendArrayTraversable[A](array: Array[A]): TraversableOnceExt[Array[A], A] =
  new TraversableOnceExt[Array[A], A](array, implicitly)

Et puis il suffit d'écrire seq.minOptionBy(_.something).

1
Jean-Philippe Pellet

Vous pouvez toujours faire quelque chose comme:

case class Foo(num: Int)

val foos: Seq[Foo] = Seq(Foo(1), Foo(2), Foo(3))
val noFoos: Seq[Foo] = Seq.empty

def minByOpt(foos: Seq[Foo]): Option[Foo] =
  foos.foldLeft(None: Option[Foo]) { (acc, elem) => 
    Option((elem +: acc.toSeq).minBy(_.num)) 
  }

Ensuite, utilisez comme:

scala> minByOpt(foos)
res0: Option[Foo] = Some(Foo(1))

scala> minByOpt(noFoos)
res1: Option[Foo] = None
0
user451151

J'ai le même problème auparavant, alors j'étend Ordered et implémente la fonction de comparaison. Voici un exemple:

 case class Point(longitude0: String, latitude0: String)  extends Ordered [Point]{

  def this(point: Point) = this(point.original_longitude,point.original_latitude)
  val original_longitude = longitude0
  val original_latitude = latitude0

  val longitude = parseDouble(longitude0).get 
  val latitude = parseDouble(latitude0).get  

  override def toString: String = "longitude: " +original_longitude +", latitude: "+ original_latitude

  def parseDouble(s: String):  Option[Double] = try { Some(s.toDouble) } catch { case _ => None }

  def distance(other: Point): Double =
    sqrt(pow(longitude - other.longitude, 2) + pow(latitude - other.latitude, 2))

 override def compare(that: Point): Int = {
  if (longitude < that.longitude)
    return -1
  else if (longitude == that.longitude && latitude < that.latitude)
    return -1
  else
    return 1
 }
}

donc si j'ai un seq de Point Je peux demander la méthode max ou min

  var points =  Seq[Point]()

val maxPoint = points.max
val minPoint = points.min
0
Dvir Arad