Come appiattire elenco di opzioni che utilizzano funzioni di ordine superiore?
-
04-10-2019 - |
Domanda
Utilizzando Scala 2.7.7:
Se ho un elenco di opzioni, posso appiattirli con un per-di comprensione:
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)
Non mi piace questo stile, e preferisco usare un HOF. Questo tentativo è troppo prolisso per essere accettabile:
scala> listOfOptions.flatMap(opt => if (opt.isDefined) Some(opt.get) else None)
res1: List[java.lang.String] = List(hi)
Intuitivamente mi sarei aspettato quanto segue al lavoro, ma non lo fa:
scala> List.flatten(listOfOptions)
<console>:6: error: type mismatch;
found : List[Option[java.lang.String]]
required: List[List[?]]
List.flatten(listOfOptions)
Anche il seguente sembra che dovrebbe funzionare, ma non si:
scala> listOfOptions.flatMap(_: Option[String])
<console>:6: error: type mismatch;
found : Option[String]
required: (Option[java.lang.String]) => Iterable[?]
listOfOptions.flatMap(_: Option[String])
^
Il meglio che posso venire in mente è:
scala> listOfOptions.flatMap(_.toList)
res2: List[java.lang.String] = List(hi)
... ma io preferirei di gran lunga non dover convertire l'opzione di una lista. Che sembra goffo.
Qualche consiglio?
Soluzione
In Scala 2.8, il lavoro si appiattisce:
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)
Questo non funziona in 2.7.7, però:
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 biblioteca collezioni è stata ridisegnata, ed è migliorato molto in 2.8, quindi forse si potrebbe desiderare di provare a utilizzare l'ultima Scala 2.8 RC e vedere se questo lo rende più facile da usare per voi.
Se proprio non si vuole utilizzare il metodo toList, Credo che si può anche scrivere in questo modo:
scala> listOfOptions.flatMap(o => o)
res: List[java.lang.String] = List(hi)
Inoltre, non una cosa di bellezza, forse, ma almeno questo funziona in 2.7.7.
Altri suggerimenti
Per completare la risposta di Arjan, a Scala 2.7.7 è possibile utilizzare List#flatten
, ma è necessario per aiutare il tipo 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)