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

Question

Я написал следующий код: Scala

case class A(x: Int, out: List[Int]) 

    def isIn: Int => List[Int] => Boolean = 
    x => l => l.filter { _ == x }.nonEmpty 

    def filterTest4: (Int, List[A]) => List[List[Int]] = (x, a) => 
    for (l <- a; if (isIn(x)(l.out))) yield l.out 

functrion filterTest4 работает прекрасно, однако использует для & выхода, который я не очень нравлюсь, и, таким образом, хотел бы видеть другой путь. Я был бы очень доволен, если бы кто-то предложил конструктивный комментарий/ответ. Пожалуйста, будьте милы и имейте в виду, я только начал писать в Scala, может быть, 3 дня назад.

Answer 1

Я только что нашел функцию карты, которая может быть использована, как это:

def filterTest5: (Int, List[A]) => List[List[Int]] = (x, a) => 
    a.filter { a2 => isIn(x)(a2.out) }.map { a2 => a2.out } 

Я думаю, что делает работу.

Answer 2

В качестве альтернативы, вы можете использовать collect, который сочетает в себе map и filter. Также вы можете разрушить класс case для прямого доступа к out.

def filterTest5: (Int, List[A]) => List[List[Int]] = (x, a) => a.collect { 
    case A(_, out) if isIn(x)(out) => out 
}