使我的数组为零，保持一些索引不变

Question

我有一个数组'board_matrix'，里面有一些值。我还有另一个数组'swin'，里面还有一些索引值。现在我想让"board_matrix"变为零，除了"win"数组中的索引。

比如

board_matrix = [1,0,2,2,1,0,1,0,1]
win = [0,4,8]
then output should be 
new_array = [1,0,0,0,1,0,0,0,1]

Answer 1

如果在win数组中找不到索引，则可以迭代board_matrix数组并覆盖这些值。类似这样的东西：

for (i=0; i<board_matrix.size; i++) {
if (!win.contains(i)) board_matrix[i] = 0;
}

Answer 2

board_matrix.enumerated().map { [winIndices = Set(win)] in
winIndices.contains($0.offset) ? $0.element : 0
}