Swift 中的高阶函数——map、compactMap、flatMap

简介

在上一篇文章中，我们了解了高阶函数的概念以及如何在 Swift 中创建自己的高阶函数。今天，我们将重点介绍 Swift 中捆绑的一些特定高阶函数，包括 map、compactMap 和 flatMap。这些函数非常有用，可以帮助我们简化和优化代码，让我们的代码更具可读性和可维护性。

map 函数

map 函数用于将一个数组中的每个元素映射到一个新的值，并返回一个包含这些新值的新数组。它接受两个参数：一个闭包和一个数组。闭包指定如何将数组中的每个元素映射到一个新值，而数组是要映射的数组。例如，以下代码使用 map 函数将一个整数数组中的每个元素加 1，并返回一个包含这些新值的新数组：

let numbers = [1, 2, 3, 4, 5]
let doubledNumbers = numbers.map { $0 + 1 }
print(doubledNumbers) // 输出：[2, 3, 4, 5, 6]

compactMap 函数

compactMap 函数与 map 函数非常相似，但它有一个关键的区别：它会忽略 nil 值。这意味着它只将数组中的非 nil 元素映射到新值，并返回一个包含这些新值的新数组。例如，以下代码使用 compactMap 函数从一个字符串数组中提取数字，并返回一个包含这些数字的新数组：

let strings = ["1", "2", "3", "a", "b", "c"]
let numbers = strings.compactMap { Int($0) }
print(numbers) // 输出：[1, 2, 3]

flatMap 函数

flatMap 函数与 map 函数和 compactMap 函数类似，但它可以处理嵌套数组。它接受两个参数：一个闭包和一个数组。闭包指定如何将数组中的每个元素映射到一个新值，而数组是要映射的数组。不同之处在于，闭包可以返回一个数组，而不是一个值。这意味着 flatMap 函数可以将嵌套数组“扁平化”，并将所有元素放入一个新的数组中。例如，以下代码使用 flatMap 函数将一个数组中的数组“扁平化”，并返回一个包含所有元素的新数组：

let arrayOfArrays = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
let flattenedArray = arrayOfArrays.flatMap { $0 }
print(flattenedArray) // 输出：[1, 2, 3, 4, 5, 6, 7, 8, 9]

何时使用这些函数

map、compactMap 和 flatMap 函数都是非常有用的工具，可以帮助我们简化和优化代码。它们可以用于各种各样的任务，包括：

• 将数组中的每个元素映射到一个新值
• 从数组中提取特定元素
• 将嵌套数组“扁平化”
• 对数组中的元素进行过滤

结论

Swift 中的高阶函数非常强大，可以帮助我们编写出更简洁、更优雅、更可维护的代码。通过熟练掌握这些函数，我们可以大大提高我们的编程效率和代码质量。