Uzyskaj wszystkie wpisy o wartości z listy [Opcja] w Scala

Question

Czy można uzyskać wszystkie wpisy o wartości List[Option[T]]?

Przykład:

val list = List(None, Some(1), None, Some(2)) 
list.filter(_.isDefined).map(_.get) 

wynik:

List[Int] = List(1, 2) 

Czy istnieje sposób, aby zrobić to w jednym kroku? To powszechny przypadek, prawda?

Answer 1

Zauważ, że

list.flatten 

zrobi tak samo dobrze.

Answer 2

Tak, można użyć collect za to:

list.collect { case Some(x) => x } 

collect zajmuje PartialFunction[A,B] i zastosowanie tej funkcji do wszystkich elementów, na których jest ona definiowana i odrzuca resztę.

edit:

Jak gpampara poprawnie wspomniano, w tym przypadku flatten będzie wystarczająca. collect byłby właściwym narzędziem, gdyby istniały dodatkowe ograniczenia i/lub transformacje.

Answer 3

Można również użyć:

list.flatMap(x => x) 

scala> val a = List(None, Some(1), None, Some(2)) 
a: List[Option[Int]] = List(None, Some(1), None, Some(2)) 

scala> a.flatMap(x => x) 
res0: List[Int] = List(1, 2) 

Dla wyjaśnienia jak/dlaczego to działa, można sprawdzić ten artykuł: http://www.brunton-spall.co.uk/post/2011/12/02/map-map-and-flatmap-in-scala/