将布尔掩码应用于 Scala 数组

Question

我正在尝试使用一个布尔值数组来选择另一个数组中的特定元素。例如：

val arr = Seq("A", "B", "C")
val mask = Seq(true,false,true)

我希望输出是一个新数组：

val arr_new = Seq("A","C")

有没有办法在 Scala 中实现这一点？

Answer 1

scala> arr.zip(mask).collect { case (v, true) => v }
res0: Seq[String] = List(A, C)