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

Использование 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)

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

Любые советы?

Ответы

Ответ 1

В Scala 2.8 сглаживание будет работать:


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.

Ответ 2

Чтобы дополнить ответ Arjan, в Scala 2.7.7 вы можете использовать List#flatten, но вам нужно помочь использовать тип 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)