0

是否可以通过 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)))))
4

2 回答 2

5

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

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

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

于 2018-09-05T07:56:08.113 回答
0
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))))
于 2018-09-05T08:05:21.163 回答