Scala: lọc một tập hợp các tùy chọn

Question

Nói rằng tôi có một chức năng để kiểm tra xem một số hoạt động được áp dụng cho một thể hiện của A và nếu như vậy, trả về một thể hiện của B hoặc None:

def checker[A,B](a: A) : Option[B] = ... 

Bây giờ tôi muốn để tạo một bộ sưu tập mới chứa tất cả các phiên bản B hợp lệ, hãy bỏ các giá trị None. Đoạn mã sau dường như thực hiện công việc, nhưng chắc chắn có một cách tốt hơn:

val as = List[A](a1, a2, a3, ...) 
    val bs = 
    as 
    .map((a) => checker(a)) // List[A] => List[Option[B]] 
    .filter(_.isDefined)  // List[Option[B]] => List[Option[B]] 
    .map(_.get)    // List[Option[B]] => List[B] 

Cảm ơn!

Answer 1

này nên làm điều đó:

val bs = as.flatMap(checker) 

Answer 2

Câu trả lời trên là chính xác, nhưng nếu bạn có thể viết lại checker, tôi đề nghị bạn sử dụng PartialFunction và collect. PartialFunction là một chức năng của loại A => B mà không cần định nghĩa cho tất cả các giá trị của A. Đây là một ví dụ đơn giản:

scala> List(1, 2, 3, 4, "5") collect {case x : Int => x + 42} 
res1: List[Int] = List(43, 44, 45, 46) 

collect mất một thể hiện của PartialFunction như là đối số và áp dụng nó cho tất cả các yếu tố của bộ sưu tập. Trong trường hợp của chúng tôi, hàm chỉ được xác định cho Ints và "5" được lọc. Vì vậy, collect là sự kết hợp của map và filter, chính xác là trường hợp của bạn.