Mapa con solo ciertas teclas

Question

Con respecto a los mapas en Scala, si ms - (k, 1, m) devuelve el mapa que contiene todas las asignaciones de ms excepto para cualquier asignación con las claves dadas, x, 1 y m.

Entonces, ¿qué enunciado devolverá un mapa de todas las asignaciones de ms con solo las teclas dadas, x, 1 y m. es decir, estoy buscando el subconjunto de ms donde solo k, 1 ym son claves.

Esto funciona, pero es terrible:

scala> val originalMap = Map("age" -> "20", "name" -> "jack", "hobby" -> "jumping") 
ms: scala.collection.immutable.Map[java.lang.String,java.lang.String] = Map(age -> 20, name -> jack, hobby -> jumping) 

scala> val interestingKeys = List("name", "hobby") 
interesting: List[java.lang.String] = List(name, hobby) 

scala> val notInterestingMap = originalMap -- interestingKeys 
notInterestingMap: scala.collection.immutable.Map[java.lang.String,java.lang.String] = Map(age -> 20) 

scala> val interestingMap = originalMap -- notInterestingMap.keySet 
interestingMap: scala.collection.immutable.Map[java.lang.String,java.lang.String] = Map(name -> jack, hobby -> jumping) 

Answer 1

Debido filterKeys filtros basados ​​en un predicado arbitraria, tiene que considerar cada clave en el mapa. Esto podría estar bien o no, dependiendo de qué tan grande sea el mapa, etc., pero definitivamente no es necesario para la operación que describes. Me gustaría usar algo como lo siguiente:

interestingKeys.flatMap(k => originalMap.get(k).map((k, _))).toMap 

Ésta será O(n) o O(n log m) dependiendo de la implementación mapa (donde n es el tamaño de interestingKeys y m es el tamaño del mapa), en lugar de o O(mn) .

Si realmente quiere que su operador de ~, puede utilizar los pimp-my-library pattern:

class RichMap[A, B](m: Map[A, B]) { 
    def ~(ks: A*) = ks.flatMap(k => m.get(k).map((k, _))).toMap 
} 

implicit def enrichMap[A, B](m: Map[A, B]) = new RichMap(m) 

ahora originalMap ~ ("name", "hobby") vuelve Map(name -> jack, hobby -> jumping), como era de esperar.

Answer 2

filterKeys pueden ayudar:

scala> originalMap.filterKeys(interestingKeys.contains) 
res0: scala.collection.immutable.Map[java.lang.String,java.lang.String] = Map(name -> jack, hobby -> jumping) 

Answer 3

creo que el código original no es tan malo, y que se puede transformar fácilmente en una sola línea que operan en el juego de llaves:

val interestingMap = originalMap -- (originalMap.keySet -- interestingKeys) 

Me parece bastante legible.