Remplacer mapValues ​​pour ListMap

Question

Scala immutable Map a un membre mapValues, qui permet d'effectuer le mappage sur des valeurs uniquement (pas les clés). Lorsque vous utilisez immutable ListMap à la place, ce membre est hérité, mais il n'est pas remplacé, donc il renvoie toujours un Map, pas ListMap.

Existe-t-il un moyen simple de mettre en œuvre mapValues pour un ListMap?

Dans le fragment suivant, je veux que le type de retour d'être un ListMap, non Map:

import scala.collection.immutable.ListMap 

val lm = ListMap(1 -> "1", 0 -> "0", 2 -> "2") 

lm.mapValues (v => v+v) 

Answer 1

Que diriez-vous:

def listMapValues[K,V1,V2](lm:ListMap[K,V1], f: V1 => V2) = lm.map { case (k,v1) => (k,f(v1))} 

Vous pouvez alors l'utiliser comme:

scala> listMapValues(lm, v => v + v) 
res16: scala.collection.immutable.ListMap[Int,String] = Map(1 -> 11, 0 -> 00, 2 -> 22) 

Si vous voulez l'utiliser comme méthode infixe, il suffit de le déclarer implicite t class:

implicit class ListMapOps[K,V1](lm: ListMap[K,V1]) { 
    def listMapValues[V2](f: V1 => V2)= lm.map { case (k,v1) => (k,f(v1))} 
    } 


scala> lm.listMapValues(v => v + v) 
res17: scala.collection.immutable.ListMap[Int,String] = Map(1 -> 11, 0 -> 00, 2 -> 22) 

Answer 2

Vous pouvez faire ceci.

ListMap[Int, String]().empty ++ lm.mapValues(v => v + v) 

Votre code:

import scala.collection.immutable.ListMap 
    val lm: ListMap[Int, String] = ListMap(1 -> "1", 0 -> "0", 2 -> "2") 
    val result: ListMap[Int, String] = ListMap[Int, String]().empty ++ lm.mapValues(v => v + v) 

Sortie à REPL:

import scala.collection.immutable.ListMap 
lm: scala.collection.immutable.ListMap[Int,String] = Map(1 -> 1, 0 -> 0, 2 -> 2) 
result: scala.collection.immutable.ListMap[Int,String] = Map(1 -> 11, 0 -> 00, 2 -> 22)