Как сгладить список опций, используя функции высшего порядка? - PullRequest
Новая
       17

Как сгладить список опций, используя функции высшего порядка?

31 голосов
/ 24 мая 2010

Использование Scala 2.7.7:

Если у меня есть список опций, я могу сгладить их, используя для понимания: 

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)

Мне не нравится этот стиль, и я предпочел бы использовать HOF. Эта попытка слишком многословна, чтобы быть приемлемой: 

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

Интуитивно я ожидал, что сработает следующее, но это не так: 

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

Даже следующее выглядит так, как будто оно должно работать, но не работает: 

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

Лучшее, что я могу придумать: 

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

... но я бы предпочел не преобразовывать опцию в список. Это кажется неуклюжим.

Любой совет?

Ответы [ 2 ]

50 голосов
/ 24 мая 2010

В Scala 2.8 будет работать flatten: 


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)

Это не работает в 2.7.7, однако: 


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

Библиотека коллекций была переработана и значительно улучшилась в 2.8, поэтому, возможно, вы захотите попробовать использовать новейшую версию Scala 2.8 RC и посмотреть, не облегчит ли это ее вам.

Если вы действительно не хотите использовать метод toList, я думаю, вы также можете написать его так: 


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

Также, возможно, это не красота, но, по крайней мере, это работает в 2.7.7.

19 голосов
/ 24 мая 2010

Чтобы дополнить ответ Арджана, в Scala 2.7.7 вы можете использовать List#flatten, но вам нужно помочь выводу типа: 

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)
...