来自数组的Scala HashMap

Question

我有一个由2元组组成的Scala数组，如下所示：

(("A", "2015-11-01"), ("B", "2016-11-11"), ("A", "2017-11-01"), ("B", "2013-11-11"))

我想创建一个地图，其中的关键字映射到最新的日期。因此，在上面的示例中，结果应该是：

Map ("A" -> "2017-11-01", "B" -> "2016-11-11")

我知道如何迭代--但是什么是Scala方式(函数式方式)来做到这一点？

Answer 1

首先选择groupBy密钥，然后选择最新日期。

arr
  .groupBy(_._1)
  .map { case (k, v) => k -> v.maxBy(_._2)._2 }

使用mapValues使其更短

arr.groupBy(_._1).mapValues(_.maxBy(_._2)._2)

由于日期(字符串)的格式正确，因此最大日期是最晚日期。您不需要将日期转换为时间来决定最大日期。

Scala

scala> val arr = Array(("A", "2015-11-01"), ("B", "2016-11-11"), ("A", "2017-11-01"), ("B", "2013-11-11"))
arr: Array[(String, String)] = Array((A,2015-11-01), (B,2016-11-11), (A,2017-11-01), (B,2013-11-11))

scala> :paste
// Entering paste mode (ctrl-D to finish)

arr
  .groupBy(_._1)
  .map { case (k, v) => k -> v.maxBy(_._2)._2 }


// Exiting paste mode, now interpreting.

res0: scala.collection.immutable.Map[String,String] = Map(A -> 2017-11-01, B -> 2016-11-11)

日期转换是不需要的，但如果你想转换它，那就继续吧。

日期转换：

//ensure correct date format is given to this method if not it will throw match error at runtime.
def convertStringDateToMillis(str: String): Long = {
 val regex = "(\\d{4})-(\\d{2})-(\\d{2})".r.unanchored
 val regex(year, month, day) = str
 val calendar = Calendar.getInstance()
 calendar.clear()
 calendar.set(Calendar.MONTH, month.toInt)
 calendar.set(Calendar.YEAR, year.toInt)
 calendar.set(Calendar.DAY_OF_MONTH, month.toInt)
 calendar.getTimeInMillis();
}

解决方案：

val arr = Array(("A", "2015-11-01"), ("B", "2016-11-11"), ("A", "2017-11-01"), ("B", "2013-11-11"))

arr.groupBy(_._1).map { case (k, v) => k -> v.maxBy(convertStringDateToMillis(_._2))._2 }

Scala

scala> def convertStringDateToMillis(str: String): Long = {
     |  val regex = "(\\d{4})-(\\d{2})-(\\d{2})".r.unanchored
     |  val regex(year, month, day) = str
     |  val calendar = Calendar.getInstance()
     |  calendar.clear()
     |  calendar.set(Calendar.MONTH, month.toInt)
     |  calendar.set(Calendar.YEAR, year.toInt)
     |  calendar.set(Calendar.DAY_OF_MONTH, month.toInt)
     |  calendar.getTimeInMillis();
     | }
convertStringDateToMillis: (str: String)Long

scala> val arr = Array(("A", "2015-11-01"), ("B", "2016-11-11"), ("A", "2017-11-01"), ("B", "2013-11-11"))
arr: Array[(String, String)] = Array((A,2015-11-01), (B,2016-11-11), (A,2017-11-01), (B,2013-11-11))


scala> arr.groupBy(_._1).map { case (k, v) => k -> v.maxBy(x => convertStringDateToMillis(x._2))._2 }
res3: scala.collection.immutable.Map[String,String] = Map(A -> 2017-11-01, B -> 2016-11-11)

Answer 2

像这样的东西应该是有效的：

array
  .groupBy(_._1)
  .mapValues(_.map(_._2).max)