Scala：如何从Set [K]和从K到V的函数创建Map [K，V]？

Question

从Set[K]创建Map[K,V] Set[K]并从K到V函数的最佳方法是什么？

例如，假设我有

scala> val s = Set(2, 3, 5)
s: scala.collection.immutable.Set[Int] = Set(2, 3, 5)

和

scala> def func(i: Int) = "" + i + i
func: (i: Int)java.lang.String

创建Map[Int, String](2 -> "22", 3 -> "33", 5 -> "55")的最简单方法是什么Map[Int, String](2 -> "22", 3 -> "33", 5 -> "55")

Answer 1

你可以使用foldLeft ：

val func2 = (r: Map[Int,String], i: Int) => r + (i -> func(i))
s.foldLeft(Map.empty[Int,String])(func2)

这将比Jesper的解决方案表现更好，因为foldLeft在一次传递中构造Map 。 Jesper的代码首先创建一个中间数据结构，然后需要将其转换为最终的Map 。

更新：我写了一个微基准测试每个答案的速度：

Jesper (original): 35s 738ms
Jesper (improved): 11s 618ms
           dbyrne: 11s 906ms
         Rex Kerr: 12s 206ms
          Eastsun: 11s 988ms

看起来它们几乎都是一样的，只要你避免构建一个中间数据结构。

Answer 2

那这个呢：

(s map { i => i -> func(i) }).toMap

这会将s的元素映射到元组(i, func(i)) ，然后将生成的集合转换为Map 。

注意： i -> func(i)与(i, func(i)) 。

dbyrne建议首先创建一个集合的视图（请参阅他的答案和评论），这样可以防止创建中间集合，从而提高性能：

(s.view map { i => i -> func(i) }).toMap

Answer 3

scala> import collection.breakOut
import collection.breakOut

scala> val set = Set(2,3,5)
set: scala.collection.immutable.Set[Int] = Set(2, 3, 5)

scala> def func(i: Int) = ""+i+i
func: (i: Int)java.lang.String

scala> val map: Map[Int,String] = set.map(i => i -> func(i))(breakOut)
map: Map[Int,String] = Map(2 -> 22, 3 -> 33, 5 -> 55)

scala>

Answer 4

除了现有的答案，

Map() ++ set.view.map(i => i -> f(i))

非常简短，表现更快（折叠/ breakOut）。

（注意视图以防止创建新集合;它会在重新映射时执行。）

Answer 5

与所有优秀的语言一样，有一百万种方法可以做任何事情。

这是一个用自己拉动集合的策略。

val s = Set(1,2,3,4,5)
Map(s.zip(s.map(_.toString)).toArray : _*)

编辑：（_。toString）可以替换为返回V类型的函数

Answer 6

其他解决方案缺乏创造力。 这是我自己的版本，虽然我真的想摆脱_.head地图。

s groupBy identity mapValues (_.head) mapValues func

Answer 7

没有使用“字符串重复”运算符*定义func（i：Int）：

scala> s map { x => x -> x.toString*2 } toMap
res2: scala.collection.immutable.Map[Int,String] = Map(2 -> 22, 3 -> 33, 5 -> 55)