scala - 从列表中过滤空值的惯用 Scala 方法

Question

我有这样的代码:

val mylist = (1,2,3)
val filteredList: List[Int] = mylist.map { num =>
   if (num < 3) num
}

所以我希望 filteredList 包含 (1,2)。然而，在这之后 filteredList 的类型是 List[Any]，而不是 List[Int]。如何删除 void/null/？值使其成为真正的整数列表？我尝试添加一个 else null 子句，然后从 filteredList 中过滤掉 null 值，但它看起来不像是好的 Scala 风格......

Answer 1

假设 myList 实际上是 List(1, 2, 3) 而不是元组 (1, 2, 3)

您应该使用filter 而不是map。您不想在 map 中放置单个 if 语句(或者通常情况下甚至是 if/else)。在这种情况下，它返回 Unit，因为 else 是 filled in with Unit由编译器。

也就是说，编译器将你的代码变成:

val filteredList: List[Int] = mylist.map { num =>
   if (num < 3) num else ()
}

这意味着你实际上有

List(1, 2, Unit) // List[Any]

改为使用:

myList.filter(_ < 3)

或长格式:

myList.filter(num => num < 3)

---

(与评论相关)您还可以使用 collect 使用模式匹配在一步中进行过滤和映射。

myList.collect { case num if(num < 3) =>
   num // or do something with num to return a different type
}