Swift中如何遍历数组的指定部分（最多前3个元素）

原代码问题分析 #

你的代码存在两个关键问题：

• 条件判断错误：if numElements = 0里用了赋值符号=，应该改为判断相等的==
• 循环逻辑混乱：while嵌套for的写法完全偏离需求，for循环会遍历整个数组的索引（还从1开始，漏掉了第一个元素），while还会重复执行三次，导致整个数组被遍历三遍

正确实现方式 #

下面提供几种简洁高效的实现，完美匹配「数组元素≥3则遍历前3个，不足3则遍历全部」的需求：

方法一：使用prefix(_:)（推荐）

Swift数组自带的prefix(_:)方法就是为这种场景设计的——它会返回数组的前n个元素，若数组元素数量不足n，则返回整个数组，代码简洁易读：

let myArray = ["Joe's","Sal's","Pizza Hut","Domino's","John's","Pizza King"] // 来自API，数量不定

if myArray.isEmpty {
    print("none found")
} else {
    // 取前3个元素，不足3个则取全部
    let targetElements = myArray.prefix(3)
    for element in targetElements {
        print(element)
    }
}

方法二：手动控制索引范围

如果想手动处理索引，可以先计算需要遍历的最大索引，再循环：

let myArray = ["Joe's","Sal's","Pizza Hut","Domino's","John's","Pizza King"]

if myArray.isEmpty {
    print("none found")
} else {
    // 前3个元素的索引是0、1、2，取数组最大索引和2的较小值
    let maxIndex = min(myArray.count - 1, 2)
    for i in 0...maxIndex {
        print(myArray[i])
    }
}

方法三：配合forEach遍历

也可以用forEach简化遍历逻辑：

let myArray = ["Joe's","Sal's","Pizza Hut","Domino's","John's","Pizza King"]

if myArray.isEmpty {
    print("none found")
} else {
    myArray.prefix(3).forEach {
        print($0)
    }
}

内容的提问来源于stack exchange，提问作者user6631314