list - Scala 分区/收集用法

Question

是否可以使用一次 collect 调用来创建 2 个新列表？如果没有，我该如何使用分区来做到这一点？

Answer 1

collect(在TraversableLike上定义并在所有子类中可用)与集合和PartialFunction一起使用。碰巧的是，大括号内定义的一堆 case 子句是一个部分函数(请参阅 Scala Language Specification 的第 8.5 节[警告 - PDF])

如异常处理:

try {
  ... do something risky ...
} catch {
  //The contents of this catch block are a partial function
  case e: IOException => ...
  case e: OtherException => ...
}

这是定义只接受给定类型的某些值的函数的便捷方法。

考虑在混合值列表上使用它:

val mixedList = List("a", 1, 2, "b", 19, 42.0) //this is a List[Any]
val results = mixedList collect {
  case s: String => "String:" + s
  case i: Int => "Int:" + i.toString
}

collect 方法的参数是 PartialFunction[Any,String]。 PartialFunction 因为它没有为 Any 类型(即 List 的类型)和 String 类型的所有可能输入定义> 因为这就是所有子句返回的内容。

如果您尝试使用 map 而不是 collect，则 mixedList 末尾的 double 值将导致 MatchError 。使用collect只会丢弃此值以及未定义 PartialFunction 的任何其他值。

一种可能的用途是将不同的逻辑应用于列表的元素:

var strings = List.empty[String]
var ints = List.empty[Int]
mixedList collect {
  case s: String => strings :+= s
  case i: Int => ints :+= i
}

尽管这只是一个示例，但许多人认为使用这样的可变变量是一种 war 犯罪 - 所以请不要这样做!

一个更好的解决方案是使用两次collect:

val strings = mixedList collect { case s: String => s }
val ints = mixedList collect { case i: Int => i }

或者，如果您确定列表仅包含两种类型的值，则可以使用 partition，它根据集合是否与某些谓词匹配将集合拆分为值:

//if the list only contains Strings and Ints:
val (strings, ints) = mixedList partition { case s: String => true; case _ => false }

这里的问题是，strings 和 ints 都是 List[Any] 类型，尽管您可以轻松地将它们强制返回某些值更加类型安全(也许通过使用 collect...)

如果您已经有一个类型安全的集合，并且想要拆分元素的其他一些属性，那么事情对您来说会更容易一些:

val intList = List(2,7,9,1,6,5,8,2,4,6,2,9,8)
val (big,small) = intList partition (_ > 5)
//big and small are both now List[Int]s

希望总结这两种方法可以帮助您!