Scala: карта + фильтр вместо foldLeft - PullRequest
Новая
       32

Scala: карта + фильтр вместо foldLeft

0 голосов
/ 05 сентября 2018

Можно ли заменить функцию foldLeft с помощью некоторой комбинации map и filter в Scala? Например на эту задачу.

Вводится список троек (имя студента, курс, оценка): 

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))))

Как добиться того же результата, используя только функции карт и фильтров? Пока у меня есть это, но выход немного отличается. 

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)))))

Ответы [ 2 ]

0 голосов
/ 05 сентября 2018 
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))))
0 голосов
/ 05 сентября 2018

Использование groupBy является хорошим способом решения этой проблемы: 

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

(Благодаря m4gic от улучшения до оригинальной версии, предложенной в комментарии)

...