Как выполнить цикл с индексом и элементом в Swift


784

Есть ли функция, которую я могу использовать для итерации по массиву и иметь индекс и элемент, как перечисление Python?

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

Ответы:


1662

Да. Начиная с 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)")
}

3
Хотя это похоже на кортеж, в Swift 1.2 - не уверен насчет 2.0 - enumerate возвращает структуру EnumerateSequence <base: SequenceType>.
15:00

2
@ Левиатлон заметные или измеримые издержки производительности, которые будут иметь значение? Нет.
Дэн Розенстарк

получить доступ к индексу единственное преимущество использования enumerate?
Мед

29
Может быть, они перейдут в герунду, enumeratingдля Swift 4. Волнующий!
Дэн Розенстарк

3
@ Мед for (index, element) inпри использовании enumeratedвводит в заблуждение. должно бытьfor (offset, element) in
Лев Дабус

116

Swift 5 предоставляет метод enumerated()для вызова Array. 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
 */

1
Является ли получение доступа к индексу единственным преимуществом использования enumerate?
Мед

52

Начиная с Swift 2, функция enumerate должна вызываться для коллекции следующим образом:

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

47

Я нашел этот ответ, когда искал способ сделать это с помощью словаря , и оказалось, что его довольно легко адаптировать, просто передавая кортеж для элемента.

// Swift 2

var list = ["a": 1, "b": 2]

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

18

Вы можете просто использовать цикл перечисления, чтобы получить желаемый результат:

Свифт 2:

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

Свифт 3 и 4:

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

Или вы можете просто пройти через цикл for, чтобы получить тот же результат:

for index in 0..<elements.count {
    let element = elements[index]
    print("\(index): \(element)")
}

Надеюсь, поможет.


14

Основной перечислять

for (index, element) in arrayOfValues.enumerate() {
// do something useful
}

или со Swift 3 ...

for (index, element) in arrayOfValues.enumerated() {
// do something useful
}

Перечислять, фильтровать и отображать

Однако я чаще всего использую перечисление в сочетании с картой или фильтром. Например, при работе с парой массивов.

В этом массиве я хотел отфильтровать нечетные или даже проиндексированные элементы и преобразовать их из Ints в Double. Так enumerate()получает индексировать и элемент, а затем проверяет фильтр индекса, и , наконец , избавиться от результирующего набора Я сопоставляет его только элемент.

let evens = arrayOfValues.enumerate().filter({
                            (index: Int, element: Int) -> Bool in
                            return index % 2 == 0
                        }).map({ (_: Int, element: Int) -> Double in
                            return Double(element)
                        })
let odds = arrayOfValues.enumerate().filter({
                            (index: Int, element: Int) -> Bool in
                            return index % 2 != 0
                        }).map({ (_: Int, element: Int) -> Double in
                            return Double(element)
                        })

9

Использование .enumerate()работает, но оно не обеспечивает истинный индекс элемента; он предоставляет только Int, начинающийся с 0 и увеличивающийся на 1 для каждого последующего элемента. Это обычно не имеет значения, но есть вероятность неожиданного поведения при использовании с ArraySliceтипом. Возьмите следующий код:

let a = ["a", "b", "c", "d", "e"]
a.indices //=> 0..<5

let aSlice = a[1..<4] //=> ArraySlice with ["b", "c", "d"]
aSlice.indices //=> 1..<4

var test = [Int: String]()
for (index, element) in aSlice.enumerate() {
    test[index] = element
}
test //=> [0: "b", 1: "c", 2: "d"] // indices presented as 0..<3, but they are actually 1..<4
test[0] == aSlice[0] // ERROR: out of bounds

Это несколько надуманный пример, и на практике это не распространенная проблема, но все же я думаю, что стоит знать, что это может произойти.


2
it does not actually provide the true index of the element; it only provides an Int beginning with 0 and incrementing by 1 for each successive elementДа, именно поэтому он называется перечислять . Кроме того, срез не является массивом, поэтому неудивительно, что он ведет себя по-другому. Здесь нет ошибки - все по дизайну. :)
Эрик Ая

5
Правда, но я никогда не называл это ошибкой. Это просто неожиданное поведение, которое, как я думал, стоит упомянуть для тех, кто не знает, как оно может негативно взаимодействовать с типом ArraySlice.
Коул Кэмпбелл

Знаете ли вы какой-либо способ получить индекс фактического элемента - например, если использовать filterсначала?
Александр Кассань


9

Для полноты вы можете просто перебрать индексы вашего массива и использовать индекс для доступа к элементу по соответствующему индексу:

let list = [100,200,300,400,500]
for index in list.indices {
    print("Element at:", index, " Value:", list[index])
}

Использование forEach

list.indices.forEach {
    print("Element at:", $0, " Value:", list[$0])
}

Используя enumerated()метод сбора . Обратите внимание, что он возвращает коллекцию кортежей с offsetи element:

for item in list.enumerated() {
    print("Element at:", item.offset, " Value:", item.element)
}

используя forEach:

list.enumerated().forEach {
    print("Element at:", $0.offset, " Value:", $0.element)
}

Те будут печатать

Элемент в: 0 Значение: 100

Элемент в: 1 Значение: 200

Элемент в: 2 Значение: 300

Элемент в: 3 Значение: 400

Элемент в: 4 Значение: 500

Если вам нужен индекс массива (не смещение) и его элемент, вы можете расширить коллекцию и создать свой собственный метод для получения индексированных элементов:

extension Collection {
    func indexedElements(body: ((index: Index, element: Element)) throws -> Void) rethrows {
        var index = startIndex
        for element in self {
            try body((index,element))
            formIndex(after: &index)
        }
    }
}

Другая возможная реализация, предложенная Алексом, заключается в архивировании индексов коллекции с ее элементами:

extension Collection {
    func indexedElements(body: ((index: Index, element: Element)) throws -> Void) rethrows {
        for element in zip(indices, self) { try body(element) }
    }
}

Тестирование:

let list =  ["100","200","300","400","500"]
list.dropFirst(2).indexedElements {
    print("Index:", $0.index, "Element:", $0.element)
}

Это будет

Индекс: 2 Элемент: 300

Индекс: 3 Элемент: 400

Индекс: 4 Элемент: 500


Кстати, вы можете реализовать enumeratedIndices, повторяя сzip(self.indices, self)
Александр - восстановить Монику

@ Alexander-ReinstateMonica for element in zip(indices, self) { try body(element) }. Кстати, мне не нравится название, которое я выбрал, indexedElementsможет лучше описать, что он делает
Лео Дабус

Ооо, я думаю, что это лучшее имя. Да, forпетля работает, но такжеzip(self.indices, self) .forEach(body)
Александр - Восстановить Монику

@ Alexander-ReinstateMonica forEachделает цикл за кадром. Я предпочитаю, чтобы все было просто: github.com/apple/swift/blob/master/stdlib/public/core/… @inlinable public func forEach( _ body: (Element) throws -> Void ) rethrows { for element in self { try body(element) } } }
Лео Дабус

7

Это формула цикла перечисления:

for (index, value) in shoppingList.enumerate() {
print("Item \(index + 1): \(value)")
}

Для более подробной информации вы можете проверить здесь .


5

Я лично предпочитаю использовать forEachметод:

list.enumerated().forEach { (index, element) in
    ...
}

Вы также можете использовать короткую версию:

list.enumerated().forEach { print("index: \($0.0), value: \($0.1)") }

4

Xcode 8 и Swift 3: массив можно перечислить с помощью tempArray.enumerated()

Пример:

var someStrs = [String]()

someStrs.append("Apple")  
someStrs.append("Amazon")  
someStrs += ["Google"]    


for (index, item) in someStrs.enumerated()  
{  
        print("Value at index = \(index) is \(item)").  
}

приставка:

Value at index = 0 is Apple
Value at index = 1 is Amazon
Value at index = 2 is Google

3

Для тех, кто хочет использовать forEach.

Swift 4

extension Array {
  func forEachWithIndex(_ body: (Int, Element) throws -> Void) rethrows {
    try zip((startIndex ..< endIndex), self).forEach(body)
  }
}

Или

array.enumerated().forEach { ... }

2

Для того, что вы хотите сделать, вы должны использовать enumerated()метод в вашем массиве :

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

-1

Мы вызвали функцию перечисления для реализации этого. подобно

for (index, element) в array.enumerate () {

// index это индексная позиция массива

// элемент является элементом массива

}

Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.