Kotlin中FlatMap与Map的用例是什么

时间:2018-08-29 13:06:05

标签: collections kotlin flatmap

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

它具有使用flatMapmap的示例

似乎两者都在做同一件事,是否有示例显示使用flatMapmap的区别?

数据类型:

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()
}

2 个答案:

答案 0 :(得分:20)

请考虑以下示例。您有一个简单的数据结构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]]
}

flatMapmap

使用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()

答案 1 :(得分:14)

这里有三个功能。 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 }
// output: [[Hero(name=Stan Lee), Hero(name=Wade Winston Wilson)], [Hero(name=Bruce Wayne), Hero(name=Diana Prince)]]

地图允许您访问{allHeroes}中的每个Universe并(在这种情况下)返回         它的英雄名单。因此输出将是一个包含两个英雄列表的列表,         每个宇宙一个。         结果是一个列表>

平面图

allHeroes.flatMap { it.heroes } 
// output: [Hero(name=Stan Lee), Hero(name=Wade Winston Wilson), Hero(name=Bruce Wayne), Hero(name=Diana Prince)]

FlatMap允许您做与地图相同的操作,从中访问两个英雄列表             两个宇宙。但是它走得更远,使返回的列表列表变得扁平             成一个列表。             结果是一个列表

扁平化

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()这两个函数的组合