如何跳过在 map scala 中附加某些条件？

Question

我有一种情况需要通过在内部应用一些过滤器来从集合创建地图，如下面的代码所示：

//Say I have a list
//I don't have to apply filter function ...

val myList = List(2,3,4,5)

val evenList = myList.map(x=>{
if ( x is even) x
else 0
}
//And the output is : List(2,0,4,0)
//The output actually needed was List(2,4) without applying filter on top like - ```myList.filter```
//I have objects instead of numbers of a case class so the output becomes :List(object1, None, object2, None)
But actual output needed was : List(object1,object2)

//更新后的场景

val basket = List(2,4,5,6)
case class Apple(name:Option[String],size:Option[Int])

val listApples: List[Apple] = basket.map(x=>{

  val r = new scala.util.Random
  val size = r.nextInt(10)

  if(x%2!=0){
    Apple(None,None)
  }
  else Apple(Some("my-apple"),Some(size))
})

当前输出：

Apple(Some(my-apple),Some(2))
Apple(Some(my-apple),Some(0))
Apple(None,None)
Apple(Some(my-apple),Some(4))

Expected was :
Apple(Some(my-apple),Some(2))
Apple(Some(my-apple),Some(0))
Apple(Some(my-apple),Some(4))

Answer 1

我不确定我是否理解正确，但为什么不直接使用 filter:

val myList = List(2,3,4,5)

myList.filter(_ % 2 == 0)

如果您想将 Filter 作为函数：

def even(n:Int) = n % 2 == 0

myList.filter(even)

问题更新后，这里是filter和collect的区别：

过滤器：

  myList
    .filter(even) 
    .map(s => Apple(Some("my-apple"),Some(s)))

收集：

myList
    .collect{ case s if(even(s)) => Apple(Some("my-apple"),Some(s))}

两者都returnList(Apple(Some(my-apple),Some(2)), Apple(Some(my-apple),Some(4)))

所以唯一的区别是您可以使用 collect.

同时执行这两个步骤

然而，对我而言，将这两个步骤分开主要更具可读性。

Answer 2

我认为 collect 最适合您的情况。它需要一个部分函数作为参数，只有当该函数匹配时，元素才会被转换并添加到结果中：

val myList = List(2,3,4,5)

case class Wrapper(i: Int)

val evenList = myList.collect{
  case x if x % 2 == 0 => Wrapper(x)
}

在这种情况下，只有 2 和 4 会被包裹在 Wrapper:

中

List(Wrapper(2), Wrapper(4))

如何跳过在 map scala 中附加某些条件？

How to skip appending some condition in map scala?

collections

scala

immutability