Scala - 如何连接地图中多个键的值

Question

我想编写一个接受List[String]和Map[String,List[Int]]的函数。
如果键存在于地图中，我想遍历字符串列表并获取值。
我已经有了这个：

def getIntList(search_keys: List[String], mapped: Map[String, List[Int]]): List[Int] =
    for ((k:String,v:List[Int]) <- mapped if search_keys.contains(k)) yield {
      v
    }

此函数不会按预期返回List[Int]，而是返回Iterable[List[Int]]。
我已经尝试在 for 循环之后变平，但这没有用。 我想我无法理解循环中返回的内容。也许有更好的方法来做到这一点？
任何帮助都非常感谢。

Answer 1

试试l.flatMap(m.get).flatten，例如

val l = List("a", "c")
val m = Map("a" -> List(1,2), "b" -> List(3,4), "c" -> List(5,6))

l.flatMap(m.get).flatten
// res0: List[Int] = List(1, 2, 5, 6)

其中m.get 返回Option[List[Int]] 所以我们实际上必须展平两次才能到达Ints