web-dev-qa-db-fra.com

Comment aplatir la liste des options en utilisant des fonctions d'ordre supérieur?

En utilisant Scala 2.7.7:

Si j'ai une liste d'options, je peux les aplatir en utilisant un pour-compréhension:

val listOfOptions = List(None, Some("hi"), None)
listOfOptions: List[Option[Java.lang.String]] = List(None, Some(hi), None)

scala> for (opt <- listOfOptions; string <- opt) yield string
res0: List[Java.lang.String] = List(hi)

Je n'aime pas ce style et préfère utiliser un HOF. Cette tentative est trop verbeuse pour être acceptable:

scala> listOfOptions.flatMap(opt => if (opt.isDefined) Some(opt.get) else None)
res1: List[Java.lang.String] = List(hi)

Intuitivement, je me serais attendu à ce que ce qui suit fonctionne, mais ce n'est pas le cas:

scala> List.flatten(listOfOptions)
<console>:6: error: type mismatch;
 found   : List[Option[Java.lang.String]]
 required: List[List[?]]
       List.flatten(listOfOptions)

Même ce qui suit semble fonctionner, mais ce n'est pas le cas:

scala> listOfOptions.flatMap(_: Option[String])
<console>:6: error: type mismatch;
 found   : Option[String]
 required: (Option[Java.lang.String]) => Iterable[?]
       listOfOptions.flatMap(_: Option[String])
                          ^

Le mieux que je puisse trouver est:

scala> listOfOptions.flatMap(_.toList)         
res2: List[Java.lang.String] = List(hi)

... mais je préférerais de loin ne pas avoir à convertir l'option en liste. Cela semble maladroit.

Aucun conseil?

35
Synesso

Dans Scala 2.8, aplatir fonctionnera:


Welcome to Scala version 2.8.0.RC2 (Java HotSpot(TM) 64-Bit Server VM, Java 1.6.0_20).
Type in expressions to have them evaluated.
Type :help for more information.

scala> val listOfOptions = List(None, Some("hi"), None)
listOfOptions: List[Option[Java.lang.String]] = List(None, Some(hi), None)

scala> listOfOptions flatten
res0: List[Java.lang.String] = List(hi)

Cela ne fonctionne pas dans 2.7.7, cependant:


Welcome to Scala version 2.7.7.final (Java HotSpot(TM) 64-Bit Server VM, Java 1.6.0_20).

scala> val listOfOptions = List(None, Some("hi"), None)
listOfOptions: List[Option[Java.lang.String]] = List(None, Some(hi), None)

scala> listOfOptions.flatten
:6: error: no implicit argument matching parameter type (Option[Java.lang.String]) => Iterable[Nothing] was found.
       listOfOptions.flatten

La bibliothèque de collections a été repensée et s'est beaucoup améliorée dans la version 2.8, alors peut-être que vous voudrez peut-être essayer d'utiliser le dernier Scala 2.8 RC et voir si cela le rend plus facile à utiliser pour vous .

Si vous ne voulez vraiment pas utiliser la méthode toList, je suppose que vous pouvez également l'écrire comme ceci:


scala> listOfOptions.flatMap(o => o)
res: List[Java.lang.String] = List(hi)

Ce n'est peut-être pas non plus beau, mais au moins cela fonctionne en 2.7.7.

54
Arjan Blokzijl

Pour compléter la réponse d'Arjan, dans Scala 2.7.7 vous pouvez utiliser List#flatten, mais vous devez aider l'inférence de type:

Welcome to Scala version 2.7.7.final (Java HotSpot(TM) 64-Bit Server VM, Java 1.6.0_20).
Type in expressions to have them evaluated.
Type :help for more information.

scala> val listOfOptions = List(None, Some("hi"), None)
listOfOptions: List[Option[Java.lang.String]] = List(None, Some(hi), None)

scala> listOfOptions.flatten[String]                   
res0: List[String] = List(hi)

scala> val x: List[String] = listOfOptions.flatten
x: List[String] = List(hi)
19
retronym