协慌网

登录 贡献 社区

如何在 Swift 中使用索引和元素迭代循环

有没有我可以用来遍历数组并具有索引和元素的函数,例如 Python 的enumerate

for index, element in enumerate(list):
    ...

答案

是的。从 Swift 3.0 开始,如果需要每个元素的索引及其值,则可以使用enumerated()方法遍历数组。它返回由索引和数组中每个项目的值组成的对对的序列。例如:

for (index, element) in list.enumerated() {
  print("Item \(index): \(element)")
}

在 Swift 3.0 之前和 Swift 2.0 之后,该函数称为enumerate()

for (index, element) in list.enumerate() {
    print("Item \(index): \(element)")
}

在 Swift 2.0 之前, enumerate是一个全局函数。

for (index, element) in enumerate(list) {
    println("Item \(index): \(element)")
}

Swift 5 为Array提供了一种称为enumerated()的方法。 enumerated()具有以下声明:

func enumerated() -> EnumeratedSequence<Array<Element>>

返回对(n,x)对的序列,其中 n 表示从零开始的连续整数,x 表示该序列的元素。


在最简单的情况下,可以将enumerated()与 for 循环一起使用。例如:

let list = ["Car", "Bike", "Plane", "Boat"]
for (index, element) in list.enumerated() {
    print(index, ":", element)
}

/*
prints:
0 : Car
1 : Bike
2 : Plane
3 : Boat
*/

但是请注意,您不限于将enumerated()与 for 循环一起使用。实际上,如果您打算将带enumerated()与 for 循环一起使用,类似于以下代码,则您做错了:

let list = [Int](1...5)
var arrayOfTuples = [(Int, Int)]()

for (index, element) in list.enumerated() {
    arrayOfTuples += [(index, element)]
}

print(arrayOfTuples) // prints [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)]

一个更快的方法是:

let list = [Int](1...5)
let arrayOfTuples = Array(list.enumerated())
print(arrayOfTuples) // prints [(offset: 0, element: 1), (offset: 1, element: 2), (offset: 2, element: 3), (offset: 3, element: 4), (offset: 4, element: 5)]

或者,您也可以将enumerated()map

let list = [Int](1...5)
let arrayOfDictionaries = list.enumerated().map { (a, b) in return [a : b] }
print(arrayOfDictionaries) // prints [[0: 1], [1: 2], [2: 3], [3: 4], [4: 5]]

而且,尽管它有一些局限性,但是forEach可以很好地替代 for 循环:

let list = [Int](1...5)
list.reversed().enumerated().forEach { print($0, ":", $1) }

/*
prints:
0 : 5
1 : 4
2 : 3
3 : 2
4 : 1
*/

通过使用enumerated()makeIterator() ,您甚至可以在Array上手动进行迭代。例如:

import UIKit
import PlaygroundSupport

class ViewController: UIViewController {

    var generator = ["Car", "Bike", "Plane", "Boat"].enumerated().makeIterator()

    override func viewDidLoad() {
        super.viewDidLoad()

        let button = UIButton(type: .system)
        button.setTitle("Tap", for: .normal)
        button.frame = CGRect(x: 100, y: 100, width: 100, height: 100)
        button.addTarget(self, action: #selector(iterate(_:)), for: .touchUpInside)
        view.addSubview(button)
    }

    @objc func iterate(_ sender: UIButton) {
        let tuple = generator.next()
        print(String(describing: tuple))
    }

}

PlaygroundPage.current.liveView = ViewController()

/*
 Optional((offset: 0, element: "Car"))
 Optional((offset: 1, element: "Bike"))
 Optional((offset: 2, element: "Plane"))
 Optional((offset: 3, element: "Boat"))
 nil
 nil
 nil
 */

从 Swift 2 开始,需要在集合上调用枚举函数,如下所示:

for (index, element) in list.enumerate() {
    print("Item \(index): \(element)")
}