Scala：map+filter 而不是 foldLeft

Question

是否可以通过 Scala 中的 map 和 filter 的某种组合来替换 foldLeft 函数？例如这个任务。

输入是三元组列表（学生姓名、课程、年级）：

val grades = List(("Hans", "db", 2.3), ("Maria", "prog1", 1.0), ("Maria", "prog2", 1.3), ("Maria", "prog3", 1.7), ("Hans", "prog2", 1.7), ("Josef", "prog1", 1.3), ("Maria", "mathe1", 1.3), ("Josef", "db", 3.3), ("Hans", "prog1", 2.0))

然后对于每个学生，应该映射他们的课程和成绩列表。使用 foldLeft 是这样的：

grades.foldLeft(Map[String, List[(String, Double)]]())((acc, e) => acc + (e._1 -> (acc.getOrElse(e._1, List()) ::: List((e._2, e._3))))).toList

输出：

List[(String, List[(String, Double)])] = List((Hans,List((db,2.3), (prog2,1.7), (prog1,2.0))), (Maria,List((prog1,1.0), (prog2,1.3), (prog3,1.7), (mathe1,1.3))), (Josef,List((prog1,1.3), (db,3.3))))

如何仅使用 map 和 filter 函数实现相同的输出？到目前为止我有这个，但输出略有不同。

grades.map(x => (x._1, List())).distinct.flatMap(x => grades.map(z => if(!x._2.contains(z._2, z._3)) (x._1, x._2 ::: List((z._2, z._3)))))

Answer 1

使用 groupBy 是解决这个问题的好方法：

grades
  .groupBy(_._1)
  .mapValues(_.map(t => (t._2, t._3)))

（感谢 m4gic 从评论中建议的改进到原始版本）

Answer 2

grades
  .map({ case (name, _, _) => name })
  .distinct
  .map(name => {
    val scores = grades
      .filter({ case (nameInternal, _, _) => name == nameInternal })
      .map({ case (_, subject, score) => (subject, score) })
    (name, scores)
  })

// res0: List[(String, List[(String, Double)])] = List((Hans,List((db,2.3), (prog2,1.7), (prog1,2.0))), (Maria,List((prog1,1.0), (prog2,1.3), (prog3,1.7), (mathe1,1.3))), (Josef,List((prog1,1.3), (db,3.3))))

Scala：map+filter 而不是 foldLeft

Scala: map+filter instead of foldLeft

scala

functional-programming

foldleft