Scala 在迭代时更新映射中的值

Question

我正在尝试在迭代时更新 map [String, WordCount] 中的值

case class WordCount(name: String,
                 id: Int,
                 var count: Double,
                 var links: scala.collection.mutable.HashMap[String, Double],
                 var ent: Double) {

    def withEnt(v: Double): WordCount = {println(s"v: $v"); copy(ent = v)}
}

var targets = mutable.HashMap[String, WordCount]()

函数 calEnt 是:

def calEnt(total: Double, links: scala.collection.mutable.HashMap[String, Double]): Double = {
    var p: Double = 0.0
    var ent: Double = 0.0
    links.map(d => {
      p = d._2 / total
      ent -= p * Math.log(p)
    })
    return ent
  }

我正在使用:

targets.map(m => m._2.withEnt(calEnt(m._2.count, m._2.links)))

用于迭代 map 并计算 ent 的新值并使用 withEnt 更新该值。我可以在控制台中输入新值，但它没有在 map 内设置。有什么方法可以做到这一点？请。

Answer 1

使用foreach:

targets.foreach(m => targets(m._1) = m._2.withEnt(calEnt(m._2.count, m._2.links)))

独立的示例:

val m = scala.collection.mutable.HashMap[Int, Int](1 -> 1, 2 -> 2)
println(m)
m.foreach(p => m(p._1) = p._2 + 1)
println(m)