是否有像 sortWith 函数一样工作的本机分组函数？

Question

有一些库（例如 Spark 和其他 Scala 扩展）具有可用的“groupWith”函数。此功能允许您将一个元素与集合的其余部分进行比较，然后使用一个或多个谓词对其进行分组。 Scala 中似乎没有为此提供任何本机功能，但它们确实具有行为相似的 sortWith 函数，但仅对项目进行排序而不是对它们进行分组。如果解释不充分，这里有一个小代码示例，应该显示我正在尝试做的事情：

val list = List(1,2,3,4,5,5)
val groupedList = list.groupWith{ (e,c) =>
    e == c
}

这是一个很简单的例子，我想做更复杂的比较比如

e + 1 == c

那么问题是有没有任何原生的 Scala 函数可以做到这一点？有什么建议或解决方法吗？

更新： 从给出的简单示例看来，我要做什么并不完全清楚，这是一个更好的示例： 假设我有一个案例类和这些对象的列表：

case class Item(num: Int, color: String)
val list = List(new Item(13, "red"), new Item(14,"red"), new Item(15, "blue"), new Item(16, "red"))

list.groupWith{ (e,c) =>
    (e.num -1 == c.num || e.num + 1 == c.num ) && e.color == c.color        
}

这应该返回如下内容：

res8: List[List[Item]] = List(List(Item(13,red), Item(14,red)), List(Item(15,blue)), List(Item(16,red)))

Answer 1

这是一个实现：

// Takes the list as a parameter, can use pimp-my-library if you want
def groupWith[A](xs: List[A], f: (A, A) => Boolean) = {
  // helper function to add "e" to any list with a member that matches the predicate
  // otherwise add it to a list of its own
  def addtoGroup(gs: List[List[A]], e: A): List[List[A]] = {
    val (before, after) = gs.span(_.exists(!f(_, e)))
    if (after.isEmpty)
      List(e) :: gs
    else
      before ::: (e :: after.head) :: after.tail
  }
  // now a simple foldLeft adding each element to the appropriate list
  xs.foldLeft(Nil: List[List[A]])(addtoGroup)
} 

groupWith(list, { (e: Item, c: Item) =>
                    (e.num - 1 == c.num || e.num + 1 == c.num) && e.color == c.color})

//| res0: List[List[groups.groups.Item]] =
//         List(List(Item(16,red)),
//              List(Item(15 ,blue)), 
//              List(Item(14,red), Item(13,red)))

Answer 2

不确定这是否是您想要的（检查我的 cmets 到您的问题），但在 GenTraversableLike 中定义的方法 groupBy List 继承（不仅是 List）。你会得到：

scala> val list = List(1,2,3,4,5,5)
list: List[Int] = List(1, 2, 3, 4, 5, 5)

scala> list.groupBy( el => el )
res0: scala.collection.immutable.Map[Int,List[Int]] = Map(5 -> List(5, 5), 1 -> List(1), 2 -> List(2), 3 -> List(3), 4 -> List(4))

scala> list.groupBy( el => el + 1 )
res1: scala.collection.immutable.Map[Int,List[Int]] = Map(5 -> List(4), 6 -> List(5, 5), 2 -> List(1), 3 -> List(2), 4 -> List(3))

基本上你需要提供从值到键的鉴别函数，你会得到Map[Key, List[Value]。

这是你想要的吗？