Question

我有：

val intsPerChar: List[(Char, List[Int])] = List(
  'A' -> List(1,2,3),
  'B' -> List(2,3)
)

我想获得一个int类型与它们所映射的字符的映射。即，我想得到：

val charsPerInt: Map[Int, List[Char]] = Map(
  1 -> List('A'),
  2 -> List('A', 'B'),
  3 -> List('A', 'B')
)

当前，我正在执行以下操作：

val numbers: List[Int] = l.flatMap(_._2).distinct
numbers.map( n =>
  n -> l.filter(_._2.contains(n)).map(_._1)
).toMap

是否有不太明确的方法？理想情况下是某种groupBy。

Answer 1

尝试

intsPerChar
  .flatMap { case (c, ns) => ns.map((_, c)) }
  .groupBy(_._1)
  .mapValues(_.map(_._2))  
// Map(2 -> List(A, B), 1 -> List(A), 3 -> List(A, B))

Answer 2

关于您是否看似可读性可能是个人喜好，但以下是另一种选择：

intsPerChar
  .flatMap(n => n._2.map(i => i -> n._1)) // List((1,A), (2,A), (3,A), (2,B), (3,B))
  .groupBy(_._1) // Map(2 -> List((2,A), (2,B)), 1 -> List((1,A)), 3 -> List((3,A), (3,B)))
  .transform { (_, v) => v.unzip._2}

最终输出为：

Map(2 -> List(A, B), 1 -> List(A), 3 -> List(A, B))

Scala：建立一对多关系

2 个答案: