ios - 在 Swift 中查找数组中的值(类似于 Excel 中的 VLOOKUP 函数)

Question

我对 Swift 3 和一般的编程语言还很陌生。我在一个数组和一个可变收入中有以下数组:

let testArray: [[Double]] = [
    [0,0],
    [1000,20.5],
    [3000,21],
    [4000,22.5],
   ]

var income: Double = 3500

我想做的是类似于 Excel 中的 VLOOKUP 函数:我想在数组的第一列(即 0、1000、3000、4000)中找到一个等于或小于我的变量的数字.在这种情况下，由于收入 = 3500，程序应返回 3000。我尝试使用 filter() 但我不知道如何使用数组中的数组。任何帮助表示赞赏。

Answer 1

您可以进行如下操作。

获取数组的第一列:

let firstColumn = testArray.map { $0[0] }
print(firstColumn) // [0.0, 1000.0, 3000.0, 4000.0]

限制那些小于或等于的元素 给定数量:

let filtered = firstColumn.filter { $0 <= income }
print(filtered) // [0.0, 1000.0, 3000.0]

获取过滤后数组的最大元素。如果元素是 按递增顺序排序，那么您可以使用 last 而不是 max():

let result = filtered.max()!
// Or: let result = filtered.last!
print(result) // 3000.0

综合起来:

let result = testArray.map { $0[0] }.filter { $0 <= income }.max()!
print(result) // 3000.0

一个可能的优化是将map和filter组合成 平面 map :

let result = testArray.flatMap { $0[0] <= income ? $0[0] : nil }.max()!
print(result) // 3000.0

这段代码假定至少有一个匹配元素， 否则 last! 或 max()! 会崩溃。如果不能保证:

if let result = testArray.flatMap( { $0[0] <= income ? $0[0] : nil }).max() {
    print(result) // 3000.0
} else {
    print("no result")
}

或者提供一个默认值(本例中为0.0):

let result = testArray.flatMap( { $0[0] <= income ? $0[0] : nil }).max() ?? 0.0
print(result) // 3000.0