Estoy buscando una forma idiomática de convertir una lista de pares donde Pair.first es una clave y Pair.second es una lista de valores. Este enfoque de procedimiento funciona, pero esperaba encontrar una forma más idiomática que no requiera crear las listas mutables directamente.
val pairs: Pair<String, List<Int>> val res = mutableMapOf<String, List<Int>>() pairs.forEach { res.getOrPut(it.first, ::mutableListOf).addAll(it.second) }Este código puede quedar envuelto en una función de extensión como la siguiente, pero no parece muy genérico:
fun <K, V> List<Pair<K, Collection<V>>>.toMultimap(): Map<K, List<V>> { var res = mutableMapOf<K, MutableList<V>>() forEach { res.getOrPut(it.first, ::mutableListOf).addAll(it.second) } return res } El uso pairs.toMap no funciona porque sobrescribe las claves del mapa con un enfoque de último en ganar. groupBy funciona se acerca, crea claves para los valores en una lista de estructura de listas.
val pairs2 = listOf( Pair("a", listOf(1, 2, 3)), Pair("b", listOf(6, 7)), Pair("a", listOf(4, 5)), Pair("b", listOf(8, 9)), ) val res = pairs2.groupBy({ it.first }, { it.second }) println(res){a=[[1, 2, 3], [4, 5]], b=[[6, 7], [8, 9]]}
Luego, es posible aplanar el mapa, pero la desventaja aquí es que es bastante ineficiente, ya que crea el doble de hashmaps y listas requeridos por clave (uno para groupby y otro para aplanar). Sí hay
val res = pairs2.groupBy({ it.first }, { it.second }).mapValues { it.value.flatten() } println(res){a=[1, 2, 3, 4, 5], b=[6, 7, 8, 9]}
Buscando ver si hay mejores enfoques para lograr esta transformación.
En lugar de groupBy , use groupingBy , que produce un Grouping . Este es un objeto intermedio en el que puede realizar todo tipo de operaciones de plegado/reducción. En tu caso:
fun <K, V> List<Pair<K, Collection<V>>>.toMultimap() = groupingBy { it.first } .fold(emptyList<V>()) { acc, (_, new) -> acc + new } Si no le gusta el hecho de que + crea demasiadas listas nuevas, puede hacer algo como esto:
groupingBy { it.first } .fold({ _, _ -> mutableListOf<V>() }) { _, acc, (_, new) -> acc.addAll(new) acc }val pairs: List<Pair<String, List<Int>>> = listOf( Pair("a", listOf(1, 2, 3)), Pair("b", listOf(6, 7)), Pair("a", listOf(4, 5)), Pair("b", listOf(8, 9)), ) val result = pairs .map { it.first }.distinct() .associateWith { first -> pairs.filter { it.first == first }.flatMap { it.second } } println(result) // {a=[1, 2, 3, 4, 5], b=[6, 7, 8, 9]}