2010-05-24 3 views
25

Utilizzando Scala 2.7.7:Come appiattire l'elenco delle opzioni utilizzando le funzioni di ordine superiore?

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 preferiscono utilizzare 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 per lavorare, ma non è così:

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 come dovrebbe funzionare, ma doesn 't:

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 preferirei non dover convertire l'opzione in elenco. Sembra goffo.

Qualche consiglio?

risposta

46

In Scala 2.8, appiattire funzionerà:


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 dovresti provare a usare l'ultimo RC di Scala 2.8 e vedere se questo lo rende più facile da usare per te.

Se davvero 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) 
non

Anche una cosa di bellezza, forse, ma almeno questo funziona in 2.7. 7.

+3

Oh, sì. Grazie! Una volta mi sono imbattuto in flatMap (o => o) ma me ne sono dimenticato prontamente. Perché flatMap (_) non funziona allo stesso modo è una cosa curiosa. – Synesso

18

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)