Как правильно удалить элементы из изменяемой карты Scala с помощью предиката? - PullRequest
14 голосов
/ 23 марта 2010

Как это сделать, не создавая новых коллекций? Есть ли что-то лучше этого?

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 ]

18 голосов
/ 23 марта 2010

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.

2 голосов
/ 18 октября 2012

Согласно изменяемой карте Scala справочная страница , вы можете удалить один элемент с помощью - = или удалить, например:

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)

Разница в том, что - = возвращает исходный объект карты, в то время как удаление возвращает Option, содержащий значение, соответствующее удаленному ключу (если он был).

Конечно, как показывают другие ответы, если вы хотите удалить много элементов в зависимости от условия, вы должны изучить сохранение, фильтрацию и т. Д.

0 голосов
/ 23 марта 2010

Если вы используете immutable.Map, в 2.7 это может быть что-то вроде:

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

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

Поскольку нет метода retain. Очевидно, что в этом случае m должен быть объявлен как var

...