使用谓词从scala可变映射中删除元素的正确方法是什么

时间:2010-03-23 14:12:06

标签: scala scala-2.8 mutable scala-collections

如何在不创建任何新馆藏的情况下做到这一点?还有比这更好的东西吗?

val m = scala.collection.mutable.Map[String, Long]("1" -> 1, "2" -> 2, "3" -> 3, "4" -> 4)
m.foreach(t => if (t._2 % 2 == 0) m.remove(t._1))
println(m)

P.S。在Scala 2.8中

3 个答案:

答案 0 :(得分:18)

retain做你想要的。在2.7:

val a = collection.mutable.Map(1->"one",2->"two",3->"three")
a: scala.collection.mutable.Map[Int,java.lang.String] = 
  Map(2 -> two, 1 -> one, 3 -> three)

scala> a.retain((k,v) => v.length < 4)   

scala> a
res0: scala.collection.mutable.Map[Int,java.lang.String] =
  Map(2 -> two, 1 -> one)

它也有效,但我认为仍然在变化,在2.8。

答案 1 :(得分:2)

根据Scala可变地图reference page,您可以使用 - =或remove删除单个元素,如下所示:

val m = scala.collection.mutable.Map[String, Long]("1" -> 1, "2" -> 2, "3" -> 3, "4" -> 4)
m -= "1" // returns m
m.remove("2") // returns Some(2)

不同之处在于 - =返回原始地图对象,而remove返回包含与删除的键对应的值的选项(如果有的话)。

当然,正如其他答案所示,如果你想根据条件删除许多元素,你应该考虑保留,过滤等。

答案 2 :(得分:0)

如果您使用的是immutable.Map,则 2.7 可能需要:

def pred(k: Int, v: String) = k % 2 == 0

m --= (m.filter(pred(_, _)).keys

由于没有retain方法可用。显然,在这种情况下,m必须声明为var