Swift - 循环遍历数组 [[Int]] 的智能方法

Question

我使用 Array of Array [[Int]] 在地图中存储坐标。 例如，我想用“1”替换所有“2”。 为此，我创建了 2 个用于定义数组空间并检查内容的 for 循环。 这不是很灵活，只适用于定义大小的数组。

private func remove2fromArray(currentArray: [[Int]]) -> [[Int]] {

    var newArray : [[Int]] = currentArray

    for section in 0...14 {
        for row in 0...19 {
            if newArray[section][row] == 2
            { newArray[section][row] = 1 }
        }
    }
    return newArray
}

有没有办法只扫描数组的内容来替换数字？

例如 currentArray.findAndReplace(find:2, replace: 1) 或类似的，适用于 [[Int]]？

Answer 1

你可以这样做：

let result = array.map { subarray in
    subarray.map { integer -> Int in
        if integer == 2 {
            return 1
        } else {
            return integer
        }
    }
}

更短：

let result = array.map {
    $0.map { integer in
        return integer == 2 ? 1 : integer
    }
}

还有一个单行：

let result = array.map { $0.map { $0 == 2 ? 1 : $0 } }

我将尝试用简单的话来解释这里发生的事情：map 的作用是它一个一个地遍历数组元素并在元素上应用一个函数。

在我们的示例中，第一个映射遍历外部数组元素，因此这里的 $0 指的是内部数组（一个接一个）。

第二个映射遍历内部数组的元素。所以内部映射中的$0 指的是内部数组元素的每个元素。