list - Kotlin - 列表过滤中的列表

Question

我有那些数据类:

data class RouteType(

            @SerializedName("type")
            val type: String,

            @SerializedName("items")
            val items: List<RouteItem>)


data class RouteItem(

            @SerializedName("id")
            val id: String,

            @SerializedName("route")
            private val route: List<DoubleArray>)

我想按类型过滤 RouteType 列表，并按 id 过滤其中的 RouteItem 列表。

我现在的代码:

// val filter: HashMap<String, List<String>>

val result = routeTypes  // List<RouteType>
                .filter { it.type in filter.keys }
                .map {
                    routeType -> routeType.items.filter { it.id in filter[routeType.type]!! }
                }

如何制作.map返回列表中包含过滤列表？或者也许还有其他方法？

编辑

谢谢，但是 flatmap不完全是我需要的，我想。 flatmap返回嵌套列表(List<RouteItem>)，但我想要 List<RouteType> .

我通过这段代码得到了它:

val result = routeTypes
                .filter { it.type in filter.keys }
                .map {
                    routeType -> RouteType(
                        routeType.type,
                        routeType.items.filter { it.id in filter[routeType.type]!! })
                }

还有其他方法可以得到吗？

Answer 1

由于您的数据是不可变的(这是一件好事)，您需要在过滤时复制它。使用 copy 使其更具扩展性:

val result = routeTypes
        .filter { it.type in filter.keys }
        .map { it.copy(items = it.items.filter { it.id in filter[routeType.type]!! }) }