在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只会产生一个列表列表。

flatMap

请注意，在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（）是前两者的组合。

Consider the following example

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)

Map

allHeroes.map { it.heroes }

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

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

Flatmap

allHeroes.flatMap { it.heroes } 

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

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

Flatten

allHeroes.map { it.heroes }.flatten() 

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

这会产生与flatMap相同的结果。所以flatMap是两个函数的组合，map {}然后flatten（）