Scala:从地图中删除任何元素并展平

时间:2013-06-19 08:09:37

标签: scala dictionary flatten

我有一张地图:

Map("key1" -> Some("value1"), "key2" -> None, "key3" -> Some("value3"))

我想删除所有None元素并展平地图。最简单的方法是什么?我只是这样找到了:

Map("key1" -> Some("value1"), "key2" -> None, "key3" -> Some("value3")).filter(_._2.nonEmpty).map(item => (item._1 -> item._2.getOrElse(Nil)))

结果是:

Map(key1 -> value1, key3 -> value3)

你知道更好的方法吗?

5 个答案:

答案 0 :(得分:30)

我使用模式匹配是:

Map("key1" -> Some("value1"), "key2" -> None, "key3" -> Some("value3")).collect {
  case (key, Some(value)) => key -> value
}
// Map(key1 -> value1, key3 -> value3)

收集合并地图+过滤器

等行为

答案 1 :(得分:12)

你可以使用for-comprehension + pattern-matching:

for((k, Some(v)) <- yourMap) yield k -> v

答案 2 :(得分:1)

我的理解用途:

val m = Map("key1" -> Some("value1"), "key2" -> None, "key3" -> Some("value3"))
for( (key,value) <- m if(value.isDefined)) yield (key,value.get)

答案 3 :(得分:1)

在地图上使用分区,就像这样,

val (flattened,_) = map.partition(_._2.isDefined)

答案 4 :(得分:0)

您也可以定义以下帮助程序,它们允许更紧凑的语法

implicit class RichPairedOptionIterableOps[A, B, Repr[_]](
    iterable: IterableOps[(A, Option[B]), Repr, Repr[(A, Option[B])]]
  ) {
    def collectWithSome: Repr[(A, B)] = iterable.collect { case (a, Some(b)) => a -> b }
    def collectWithNone: Repr[A] = iterable.collect { case (a, None) => a }
}

以您的示例为例:

Map("key1" -> Some("value1"), "key2" -> None, "key3" -> Some("value3")).collectWithSome