ios - 基于 bool 值比较两个数组和匹配元素

Question

假设我有两个数组

let array1 = [1,2,3]
let array2 = [false, true, false]

我想从 array1 返回在特定索引处匹配 true bool 值的值列表。在这个简单的示例中，这将是 [2]

如何使用函数式方法实现？

Answer 1

“压缩”数组，然后将每一对映射到第一个元素或nil，取决于第二个元素。 flatMap() 只返回非零结果:

let array1 = [1,2,3]
let array2 = [false, true, false]

let result = zip(array1, array2).flatMap { $1 ? $0 : nil }
print(result) // [2]