filter - 如何在arrayList过滤器中获取索引列表

Question

我要获取索引列表如下。

val a = booleanArrayOf(true,false,true,false)

上面的代码，真数字是2。 -> indexList = {0，2}
如何在Kotlin中获取indexList。

Answer 1

您可以使用mapIndexed()来获取每个元素的索引和值，转换为索引或null，然后删除null。

val b: List<Int> = a.mapIndexed { i, b -> if (b) i else null }.filterNotNull().toList()

另一种方法是使用 withIndex()函数，过滤为true的值，并将结果对映射到索引值。这可能会更清楚一些。

val c: List<Int> = a.withIndex().filter { it.value }.map { it.index }