在 Scala 中转换地图

Question

我想在 Scala 中转换 String -> List[String] 类型的映射，使得每个 (k, (List v1, v2, ... vn)) 键值对都指向 {(k:v1, k), (k:v2, k), ... (k:vn, k)} 对。

比如我想改造

scala.collection.immutable.Map[String,List[String]] = Map(A -> List(a1, a2), B -> List(b1))

到

scala.collection.immutable.Map[String,List[String]] = Map(A:a1 -> A, A:a2 -> A, B:b1 -> B)

我能够使用 [this answer][1] 实现我的目标：

scala> val schema = Map("A" -> List("a1", "a2"),  "B" -> List("b1"))
schema: scala.collection.immutable.Map[String,List[String]] = Map(A -> List(a1, a2), B -> List(b1))

scala>  schema flatten {case(k, vs) => vs.map((_, k))}
res1: scala.collection.immutable.Iterable[(String, String)] = List((a1,A), (a2,A), (b1,B))

当我尝试在每个值前面加上原始键和冒号时，我得到一个错误：

scala> schema flatten {case(k, vs) => vs.map((k.concat(":").concat(_), k))}
<console>:13: error: type mismatch;
 found   : (String => String, String)
 required: String => ?
       schema flatten {case(k, vs) => vs.map((k.concat(":").concat(_), k))}

Answer 1

1. (k.concat(":").concat(k), k) 中没有 _ 和变量，所以它不是函数字面量，但 map 的参数必须是函数。
2. 无论如何，您都不想将k 与k 连接

试试这个：

schema.flatMap { case (k, vs) => vs.map(v => (k + ":" + v, k)) }

或者更简洁：

for ((k, vs) <- schema; v <- vs) yield (k + ":" + v, k)

---

编辑

对于每个k，表达式k.concat(":").concat(_) 是一个接受字符串s 并计算k + ":" + s 的函数。

因此，(k.concat(":").concat(_), k) 是一个元组

1. 一个函数x => k.concat(":").concat(x)和
2. 值k

所以它的类型为(String => String, String)。这与String => (String, String) 不同，这仍然不是您想要的。

Answer 2

错误消息告诉您.map 采用String => ? 形式的函数，但您传递给它的是(String, String) 的元组。

(k.concat(":").concat(k), k) 不是一个函数 - 我认为您并不是要在 concat 中输入两次 k。

要创建地图，您需要一个带有签名String => (String, String)的函数

• vs.map((k.concat(":").concat(k),k)) 错误地将 (String, String) 传递给映射
• vs.map((k.concat(":").concat(_),k)) 错误地将 (String => String, String) 传递给映射。
• vs.map(v => (k.concat(":").concat(v), k)) 正确匹配 String => (String, String)。

我们可以得到：

schema flatten {case(k, vs) => vs.map(v => (k.concat(":").concat(v), k)) }
 //  List((A:a1,A), (A:a2,A), (B:b1,B)) 

您可以在here 看到此代码。

但是，正如 Andrey 出色的评论所指出的，将 flatten 与隐式函数一起使用并不是一个好习惯。此外，您需要Map 而不是List。由于这两个原因，我认为你应该使用flatMap 而不是flatten：

schema flatMap {case(k, vs) => vs.map(v => (k.concat(":").concat(v), k)) };
// Map(A:a1 -> A, A:a2 -> A, B:b1 -> B)

如果您想试用，请参阅here。