时间戳

首页 > 解决方案 > 在 kotlin 中映射一个可迭代的 1-to-n

kotlin - 在 kotlin 中映射一个可迭代的 1-to-n

问题描述

我想要做的是使用Iterable.map但不是将每个值转换为一个新值，而是将一个值转换为多个新值。

例如：

val myList = listOf("test", "123", "another.test", "test2")
val result = myList.map {
    if(it.contains(".")) {
        return@map it.split(".")
    } else {
        return@map it
    }
}
//desired output: ["test", "123", "another", "test", "test2"]

此代码将生成一个包含字符串和字符串列表（类型Any）的列表。

我怎样才能最优雅地实现这一点？

标签： kotliniterable

解决方案

一种快速的方法是使用flatMap.

val output = myList.flatMap { if(it.contains(".")) it.split(".") else listOf(it) }

该flatMap方法使用给定函数转换每个元素，然后将结果展平为单个列表。

https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/flat-map.html

推荐阅读