如果键存在或不插入新元素，则在 Scala 映射中向元素添加数字的好方法

Question

我知道几个类似的问题。他们对我没有帮助 - 如果没有现有密钥，代码将无法工作。

我只需要一些不错的方法来附加 Map 并将其添加到现有键（如果它确实存在）或作为新键（如果映射不包含适当的键）。

以下代码有效，但我不喜欢它：

val a = collection.mutable.Map(("k1" -> 1), ("k2" -> 5))
val key = "k1"

val elem = a.get(key)
if (elem == None) {
    a += ("k5" -> 200)
} else {
    a.update(key, elem.get + 5)
}

有什么更好的点吗？ 当前的 Scala 版本是 2.10.4，我目前无法切换到 2.11。 可变映射不是 100% 的限制，而是首选。

例如，这里是similar question，但我还需要考虑不存在的不存在密钥的情况。至少我们应该理解a.get(key) 可以是None 或者添加一些更好的方法。 |+| 是个好主意，但我想保留基本的 Scala 2.10.x。

Answer 1

这样做的最短方法：

a += a.get(key).map(x => key -> (x + 5)).getOrElse("k5" -> 200)

一般：

a += a.get(k).map(f).map(k -> _).getOrElse(kv)

如果您的字典是不可变的，则相同：

m + m.get(k).map(f).map(k -> _).getOrElse(kv)

所以我看不出有任何理由在这里使用可变集合。

如果你不喜欢Option.map 的所有这些东西：

m + (if (m.contains(k)) k -> f(m(k)) else kv)

请注意，存在一整类可能的变化：

k1 -> f(m(k1)) else k2 -> v2 //original
k1 -> f(m(k1)) else k1 -> v2
k1 -> f(m(k2)) else k2 -> v2
k1 -> f(m(k2)) else k1 -> v2
k2 -> v2 else k1 -> f(m(k1)) 
k1 -> v2 else k1 -> f(m(k1))
k2 -> v2 else k1 -> f(m(k2))
k1 -> v2 else k1 -> f(m(k2))
... //v2 may also be a function from some key's value

那么，为什么它不是标准功能？ IMO，因为所有变体仍然可以作为单行实现。如果您想要具有所有功能的库，可以将其实现为单行，您知道，它是 Scalaz :)。

附：如果你还想知道为什么没有“update(d) if persist”功能 - 请参阅@Rex Kerr 的回答 here

Answer 2

您可以为此创建自己的函数：

def addOrUpdate[K, V](m: collection.mutable.Map[K, V], k: K, kv: (K, V), 
                      f: V => V) {
  m.get(k) match {
    case Some(e) => m.update(k, f(e))
    case None    => m += kv
  }
}

addOrUpdate(a, "k1", "k5" -> 200, (v: Int) => v + 5)

Answer 3

Scala 2.13 introduced updatedWith method 这似乎是根据密钥的存在有条件地更新地图的最惯用方式。

val a = Map(("k1" -> 1), ("k2" -> 5))

val a1 = a.updatedWith("k1") { 
  case Some(v) => Some(v + 5)
  case None => Some(200) 
}
println(a1) // Map(k1 -> 6, k2 -> 5)

也可以使用它删除值：

val a2 = a.updatedWith("k2") { 
  case Some(5) => None
  case v => v 
}
println(a2) // Map(k1 -> 1)

摘自Scala Standard Library reference：

def updatedWith[V1 >: V](key: K)(remappingFunction: (Option[V]) => Option[V1]): Map[K, V1]

更新指定键的映射及其当前可选映射的值（Some 如果有当前映射，None 如果没有）。

如果重映射函数返回Some(v)，则映射将更新为新值v。如果重新映射函数返回 None，则删除映射（如果最初不存在，则保持不存在）。如果函数本身抛出异常，则重新抛出异常，并且当前映射保持不变。

Answer 4

一种比较清晰的方法：

val a = collection.mutable.Map[String, Int]() withDefault insertNewValue

def insertNewValue(key: String): Int =
  a += key -> getValueForKey(key)
  a(key)
}

def getValueForKey(key: String): Int = key.length

不过，我完全不鼓励使用可变集合。最好将内部可变状态保留为包含不可变字段的变量。

这是因为有一条简单的规则，除非完全有必要，否则您不应该暴露您的内部状态，如果这样做，您应该尽可能减少可能产生的潜在副作用。

如果您公开一个可变状态的引用，任何其他参与者都可以更改它的值，从而失去引用透明度。 所有对可变集合的引用都很长且难以使用，这并非偶然。这是开发者给你的信息

毫不奇怪，代码仍然保持不变，在地图实例化时有一些微小的变化。

var a = Map[String, Int]() withDefault insertNewValue

def insertNewValue(key: String): Int = {
  a += key -> getValueForKey(key)
  a(key)
}

def getValueForKey(key: String): Int = key.length 

Answer 5

对于你的可变地图，你可以这样做：

val a = collection.mutable.Map(("k1" -> 1), ("k2" -> 5))
val key = "k1"

a.update(key, a.getOrElse(key, 0) + 5)

a  // val res1: scala.collection.mutable.Map[String,Int] = HashMap(k1 -> 6, k2 -> 5)