在kotlin中flatMap与地图的用例是什么？

Question

在https://try.kotlinlang.org/#/Kotlin%20Koans/Collections/FlatMap/Task.kt

它有使用flatMap和的样本map

似乎两者都在做同样的事情,是否有一个样本来显示使用flatMap和map？的区别？

数据类型:

data class Shop(val name: String, val customers: List<Customer>)

data class Customer(val name: String, val city: City, val orders: List<Order>) {
    override fun toString() = "$name from ${city.name}"
}

data class Order(val products: List<Product>, val isDelivered: Boolean)

data class Product(val name: String, val price: Double) {
    override fun toString() = "'$name' for $price"
}

data class City(val name: String) {
    override fun toString() = name
}

样品:

fun Shop.getCitiesCustomersAreFrom(): Set<City> =
    customers.map { it.city }.toSet()
    // would it be same with customers.flatMap { it.city }.toSet() ?

val Customer.orderedProducts: Set<Product> get() {
    return orders.flatMap { it.products }.toSet()
    // would it be same with return orders.map { it.products }.toSet()
}

Answer 1

请考虑以下示例.您有一个简单的数据结构,Data其中包含一个类型的属性List.

class Data(val items : List<String>)

fun main(args: Array<String>) {
    val data = listOf(Data(listOf("a", "b", "c")), Data(listOf("1", "2", "3")))
    val items: List<String> = data.flatMap { it.items } //[a, b, c, 1, 2, 3]
    val items2: List<List<String>> = data.map { it.items } //[[a, b, c], [1, 2, 3]]
}

flatMap 与 map

使用flatMap,您可以将多个"展平" Data::items为一个集合,如items变量所示.使用map,而另一方面,简单地导致列表的列表.

flatten

请注意,有一个flatten上扩展Iterable<Iterable<T>>,并且还Array<Array<T>>您可以交替使用flatMap:

val nestedCollections: List<Int> = listOf(listOf(1,2,3), listOf(5,4,3)).flatten()

Answer 2

这里有三个功能.map(),flatten()和flatMap()是前两者的组合.

请考虑以下示例

data class Hero (val name:String)
data class Universe (val heroes: List<Hero>)

val batman = Hero("Bruce Wayne")
val wonderWoman = Hero (name = "Diana Prince")

val mailMan = Hero("Stan Lee")
val deadPool = Hero("Wade Winston Wilson")

val marvel = Universe(listOf(mailMan, deadPool))
val dc = Universe(listOf(batman, wonderWoman))

val allHeroes: List<Universe> = listOf(marvel, dc)

地图

allHeroes.map { it.heroes }

Map允许您访问{allHeroes}中的每个Universe,并且(在本例中)返回其英雄列表.因此输出将是一个包含两个英雄列表的列表,每个宇宙一个.结果是List>

Flatmap

// output: [[Hero(name=Stan Lee), Hero(name=Wade Winston Wilson)], [Hero(name=Bruce Wayne), Hero(name=Diana Prince)]]

FlatMap允许您像地图一样执行,从两个Universe访问两个英雄列表.但它更进一步,并将返回的列表列表展平为单个列表.结果是一个List

弄平

allHeroes.flatMap { it.heroes } 

这会产生与flatMap相同的结果.所以flatMap是两个函数的组合,map {}然后flatten()