映射Scala映射的键和值


89

斯卡拉的MapLike特质有一种方法

mapValues [C] (f: (B) ⇒ C): Map[A, C] 

但是我有时想要一个不同的类型:

mapKeysAndValues [C] (f: (A, B) ⇒ C): Map[A, C] 

有没有一种简单的方法可以做到这一点呢?当然,这可以折叠完成。

Answers:


164

map方法遍历所有(key, value)对。您可以像这样使用它:

val m = Map("a" -> 1, "b" -> 2)

val incM = m map {case (key, value) => (key, value + 1)}

8
如果您已经具有功能f : (A,B) => (A,C),则可以简单地进行操作m.map(f.tupled)。适用于,val f = (x: String, y: Int) => (x, y+1)但奇怪的是,如果我用f等效地定义,repl会抱怨def
丹·伯顿

2
关键字case在这里能实现什么?
Omnipresent

{case (key, value) => ...}在这种情况下,@ Omnipresent 只是模式匹配。map我没有给a 提供功能,而是给它提供了部分功能。
tenshi

注意:case可以将类型放入模式中,但这并不安全,因为它可能会导致运行时错误(因为这只是模式匹配)。同时,如果您曾经更改map函数下方的集合结构,以致于对象解释的对象太少或太多(key, value),那么我再次希望您会遇到运行时错误。:(
combinatorist

8

这段代码呢:

val m = Map(1 -> "one", 2 -> "two")
def f(k: Int, v: String) = k + "-" + v
m map {case (k, v) => (k, f(k, v))}

产生:

 Map(1 -> 1-one, 2 -> 2-two)

可以打包成实用程序方法:

def mapKeysAndValues[A,B,C](input: Map[A,B], fun: (A, B) => C) = 
  input map {case(k,v) => (k, fun(k, v))}

用法:

mapKeysAndValues(
  Map(1 -> "one", 2 -> "two"), 
  (k: Int, v: String) => k + "-" + v
)

这不一样MapLike#transform吗?
Hosam Aly


2

与一些Scalaz:

scala> def fst[A, B] = (x: (A, B)) => x._1
fst: [A, B]=> (A, B) => A

scala> Map(1 -> "Lorem", 2 -> "Ipsum").map(fst &&& Function.tupled(_.toString + _))
res1: scala.collection.immutable.Map[Int,java.lang.String] = Map(1 -> 1Lorem, 2 -> 2Ipsum)

我更喜欢@tenshi的解决方案。


0

您可以创建一个实用程序类:

class MyMapLike[K,V](m:MapLike[K,V,_]){
 def mapKeysAndValues[R](f: (K, V) => R)={
   m.map{case (k,v)=> f(k,v)}
 } 
}
object MyMapLike{
 implicit def maplike2mymaplike[K,V](ml:MapLike[K,V,_]):MyMapLike[K,V]=new MyMapLike(m)

}

import MyMapLike._
Map(1 -> "one", 2 -> "two").mapKeysAndValues(k,v=>v*k)

代码未经测试,但应该可以类似地工作。

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.