Scala如何获取Map的键，其中值为List

Question

我有这样的地图：

val myMap = Map(
    testKey1 -> List(testValue1, testValue2, testValue3....), 
    testKey2 -> List(testValue4, testValue5, testValue6....),
    testKey3 -> List(testValue7, testValue8, testValue9....)
)

我想做一些list值的精确匹配，得到map对应的key。

例如：我想检查 'testValue9' 是否在此 Map 中，然后我会得到键 'testKey3'。

我认为它可以通过这种方式解决，但我无法遍历列表的值来检查该值是否存在。 或者有人可以给我一些提示。

myMap foreach {
          case (key, List(_)) => println( key )
          case _ =>    
        }

Answer 1

如果您尝试在 Map 中查找单个值，您可以使用 find：

  myMap.find(_._2.contains(value)).map(_._1)

如果找到值，则返回Some(key)，否则返回None。

如果您认为可能有多个匹配值，可以将find替换为filter

  myMap.filter(_._2.contains(value)).map(_._1)

或使用collect

  myMap.collect{ case (k, v) if v.contains(value) =>  k }

在这两种情况下，这将返回所有匹配键的列表，并取决于Map 的实现方式。

请注意，filter 选项可以使用 for 表达式来表达，其作用完全相同：

  for { (k,v) <- myMap if v.contains(value) } yield k

虽然collect 的性能可能最好，但在大多数情况下，这只是一个风格问题，即哪个更好。

更新

Raman Mishra 有用地指出过滤器版本可以简化为

myMap.filter(_._2.contains(value)).keys

这会返回Set 而不是List，后者更合适，因为顺序并不重要。

Answer 2

我想你想这样做：

假设 Key 是 String 并且 value 是 List[String]

val keyContainsValue: immutable.Iterable[String] = myMap map {
    case (key, value) => if (value.contains(testValue9)) key else ""
  }

你可以使用空字符串作为默认值，这样你就可以得到返回类型为iterable[String]。

由于我不知道您的键和值的类型，您也可以使用选项。为此目的。

val keyContainsValue: immutable.Iterable[Option[String]] = myMap map {
        case (key, value) => if (value.contains(testValue9)) Some(key) else None
      }
println(keyContainsValue.flatten) //you will get the list of keys which contains the value specified.

Answer 3

val searchValue = "testValue9"

myMap.collect{ case (key, values) if values.contains(searchValue) => key }

Answer 4

你可以做类似的事情

val getKeys = (k: String) => for (m<- myMap.keys;
                                 v<- myMap(m);
                                 if v==k) yield m