Comment aplatir la liste des options à l'aide des fonctions d'ordre supérieur?
-
04-10-2019 - |
Question
Utilisation Scala 2.7.7:
Si j'ai une liste d'options, je peux les aplatir à l'aide d'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 ne aime pas ce style, et préfère utiliser un HOF. Cette tentative est trop verbeux 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 qui suit au travail, mais il ne fonctionne pas:
scala> List.flatten(listOfOptions)
<console>:6: error: type mismatch;
found : List[Option[java.lang.String]]
required: List[List[?]]
List.flatten(listOfOptions)
Même les éléments suivants semble que cela devrait fonctionner, mais ne fait pas:
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 peux trouver est:
scala> listOfOptions.flatMap(_.toList)
res2: List[java.lang.String] = List(hi)
... mais je préférerais ne pas avoir à convertir l'option d'une liste. Cela semble maladroit.
Un conseil?
La solution
Dans Scala 2.8, aplatissez 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 en 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é, et a beaucoup amélioré 2,8, alors peut-être que vous pourriez vouloir essayer d'utiliser la dernière Scala 2.8 RC et voir si cela fait plus facile à utiliser pour vous.
Si vous ne voulez vraiment pas utiliser la méthode toList, je suppose que vous pouvez l'écrire comme ceci:
scala> listOfOptions.flatMap(o => o)
res: List[java.lang.String] = List(hi)
pas non plus une chose de la beauté peut-être, mais au moins cela fonctionne dans 2.7.7.
Autres conseils
Pour compléter la réponse de Arjan, à Scala 2.7.7 vous pouvez utiliser List#flatten
, mais vous avez besoin pour le type inferencer:
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)