5

我想实现以下目标：

(_ : Map[K,Int]).mapKey(k, _ + 1)

并且该mapKey函数仅将其第二个参数 ( Int => Int) 应用于存储在 下的值k。标准库中有什么东西吗？如果不是，我敢打赌 Scalaz 中会有一些东西。

当然，我可以自己编写这个函数（m.updated(k,f(m(k))）并且这样做很简单。但是我已经多次遇到这个问题，所以也许它已经完成了？

对于 Scalaz，我想像下面的代码：

(m: Map[A,B]).project(k: A).map(f: B => B): Map[A,B]
4

3 回答 3

3

你当然可以添加

def changeForKey[A,B](a: A, fun: B => B): Tuple2[A, B] => Tuple2[A, B] = { kv =>
  kv match {
    case (`a`, b) => (a, fun(b))
    case x => x
  }
}

val theMap = Map('a -> 1, 'b -> 2)
theMap map changeForKey('a, (_: Int) + 1)
res0: scala.collection.immutable.Map[Symbol,Int] = Map('a -> 2, 'b -> 2)

但这会规避有关内存重用和访问的任何优化。

对于您提出的方法，我还提出了一个相当冗长且效率低下的 scalaz 解决方案project：

theMap.toStream.toZipper.flatMap(_.findZ(_._1 == 'a).flatMap(elem => elem.delete.map(_.insert((elem.focus._1, fun(elem.focus._2)))))).map(_.toStream.toMap)

或者

(for {
  z <- theMap.toStream.toZipper
  elem <- z.findZ(_._1 == 'a)
  z2 <- elem.delete
} yield z2.insert((elem.focus._1, fun(elem.focus._2)))).map(_.toStream.toMap)

估计用处不大。我只是发帖供参考。

于 2012-01-11T16:19:08.423 回答
3

这是一种方法：

scala> val m = Map(2 -> 3, 5 -> 11)
m: scala.collection.immutable.Map[Int,Int] = Map(2 -> 3, 5 -> 11)

scala> m ++ (2, m.get(2).map(1 +)).sequence
res53: scala.collection.immutable.Map[Int,Int] = Map(2 -> 4, 5 -> 11)

scala> m ++ (9, m.get(9).map(1 +)).sequence
res54: scala.collection.immutable.Map[Int,Int] = Map(2 -> 3, 5 -> 11)

这是有效的，因为(A, Option[B]).sequence给Option[(A, B)]. （sequence通常将类型从里到外。即F[G[A]] => [G[F[A]]，给定F : Traverse和G : Applicative。）

于 2012-02-12T20:32:55.857 回答
1

您可以用它来拉皮条，以便它根据旧地图创建一个新地图：

class MapUtils[A, B](map: Map[A, B]) {
  def mapValueAt(a: A)(f: (B) => B) = map.get(a) match {
    case Some(b) => map + (a -> f(b))
    case None => map
  }
}

implicit def toMapUtils[A, B](map: Map[A, B]) = new MapUtils(map)

val m = Map(1 -> 1)
m.mapValueAt(1)(_ + 1)
// Map(1 -> 2)
m.mapValueAt(2)(_ + 1)
// Map(1 -> 1)
于 2012-01-12T04:56:41.170 回答