Como iterar um loop com índice e elemento no Swift


784

Existe uma função que eu possa usar para iterar sobre uma matriz e ter índice e elemento, como o enumerado do Python?

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

Respostas:


1662

Sim. No Swift 3.0, se você precisar do índice para cada elemento junto com seu valor, poderá usar o enumerated()método para iterar sobre a matriz. Retorna uma sequência de pares composta pelo índice e o valor para cada item na matriz. Por exemplo:

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

Antes do Swift 3.0 e depois do Swift 2.0, a função era chamada enumerate():

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

Antes do Swift 2.0, enumerateera uma função global.

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

3
Embora pareça uma tupla, no Swift 1.2 - não tenho certeza sobre o 2.0 - o enumerado retorna uma estrutura EnumerateSequence <base: SequenceType>.
Nstein

2
@Despesas gerais visíveis ou mensuráveis ​​do desempenho do Leviathlon que serão importantes? Não
Dan Rosenstark

obter acesso ao índice é o único benefício de usar enumerate?
Mel

29
Talvez eles mudem para o gerúndio, enumeratingpara Swift 4. Emocionante!
Dan Rosenstark 22/03

3
@Mel for (index, element) inao usar enumeratedé enganoso. deveria serfor (offset, element) in
Leo Dabus

116

Swift 5 fornece um método chamado enumerated()para Array. enumerated()tem a seguinte declaração:

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

Retorna uma sequência de pares (n, x), em que n representa um número inteiro consecutivo iniciando em zero ex representa um elemento da sequência.


Nos casos mais simples, você pode usar enumerated()com um loop for. Por exemplo:

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

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

Observe, no entanto, que você não está limitado ao uso enumerated()com um loop for. De fato, se você planeja usar enumerated()com um loop for para algo semelhante ao código a seguir, está fazendo algo errado:

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)]

Uma maneira mais rápida de fazer isso é:

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)]

Como alternativa, você também pode usar enumerated()com 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]]

Além disso, embora tenha algumas limitações , forEachpode ser um bom substituto para um loop for:

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

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

Usando enumerated()e makeIterator(), você pode até iterar manualmente no seu Array. Por exemplo:

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
Obter acesso ao índice é o único benefício de usar enumerate?
Mel

52

Começando com o Swift 2, a função enumerar precisa ser chamada na coleção da seguinte maneira:

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

47

Encontrei esta resposta enquanto procurava uma maneira de fazer isso com um dicionário , e é muito fácil adaptá-la, basta passar uma tupla para o elemento.

// Swift 2

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

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

18

Você pode simplesmente usar o loop de enumeração para obter o resultado desejado:

Swift 2:

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

Swift 3 e 4:

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

Ou você pode simplesmente passar por um loop for para obter o mesmo resultado:

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

Espero que ajude.


14

Enumerar básico

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

ou com Swift 3 ...

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

Enumerar, filtrar e mapear

No entanto, na maioria das vezes eu uso enumerar em combinação com mapa ou filtro. Por exemplo, com a operação em duas matrizes.

Nesta matriz, eu queria filtrar elementos ímpares ou pares e convertê-los de Ints para Doubles. Então enumerate()você obtém o índice e o elemento, o filtro verifica o índice e, finalmente, para se livrar da tupla resultante, mapeio-o apenas para o elemento.

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

Usando .enumerate()obras, mas não fornece o verdadeiro índice do elemento; ele fornece apenas um início Int com 0 e incrementando 1 para cada elemento sucessivo. Isso geralmente é irrelevante, mas existe o potencial de comportamento inesperado quando usado com o ArraySlicetipo. Pegue o seguinte código:

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

É um exemplo um tanto artificial e não é um problema comum na prática, mas ainda acho que vale a pena saber que isso pode acontecer.


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 elementSim, é por isso que é chamado enumerar . Além disso, a fatia não é matriz, portanto, nenhuma surpresa se comporta de maneira diferente. Não há bug aqui - tudo é por design. :)
Eric Aya

5
É verdade, mas nunca chamei isso de bug. É apenas um comportamento potencialmente inesperado que pensei que valeria a pena mencionar para aqueles que não sabiam como ele poderia interagir negativamente com o tipo ArraySlice.
Cole Campbell

Você conhece alguma maneira de obter o índice do elemento real - por exemplo, se estiver usando filterprimeiro?
Alexandre Cassagne

9

Começando com o Swift 3, é

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

9

Para garantir a integridade, você pode simplesmente percorrer os índices da matriz e usar subscrito para acessar o elemento no índice correspondente:

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

Usando o forEach

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

Usando o enumerated()método de coleção . Observe que ele retorna uma coleção de tuplas com oe offseto element:

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

usando forEach:

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

Aqueles imprimirão

Elemento em: 0 Valor: 100

Elemento em: 1 Valor: 200

Elemento em: 2 Valor: 300

Elemento em: 3 Valor: 400

Elemento em: 4 Valor: 500

Se você precisar do índice da matriz (não do deslocamento) e de seu elemento, poderá estender a Collection e criar seu próprio método para obter os elementos indexados:

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)
        }
    }
}

Outra implementação possível, como sugerido por Alex, é compactar os índices de coleção com seus elementos:

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

Teste:

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

Isso irá

Índice: 2 Elemento: 300

Índice: 3 Elemento: 400

Índice: 4 Elemento: 500


BTW você pode implementar enumeratedIndicesfazendo loop comzip(self.indices, self)
Alexander - Reinstate Monica

@ Alexander-ReinstateMonica for element in zip(indices, self) { try body(element) }. Btw eu não gosto do nome que eu escolhi, indexedElementspode descrever melhor o que faz
Leo Dabus 06/02

Acho que esse é um nome melhor. Sim, um forloop funciona, mas tambémzip(self.indices, self) .forEach(body)
Alexander - Reinstate Monica

@ Alexander-ReinstateMonica forEachfaz um loop for nos bastidores. Prefiro mantê-lo simples e simples 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) } } }
Leo Dabus

7

Esta é a fórmula do loop da enumeração:

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

para mais detalhes você pode conferir aqui .


5

Pessoalmente, prefiro usar o forEachmétodo:

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

Você também pode usar a versão curta:

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

4

Xcode 8 e Swift 3: a matriz pode ser enumerada usando tempArray.enumerated()

Exemplo:

var someStrs = [String]()

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


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

console:

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

3

Para quem quer usar forEach.

Swift 4

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

Ou

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

2

Para o que você deseja fazer, use o enumerated()método em sua matriz :

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

-1

Chamamos função enumerar para implementar isso. gostar

para (índice, elemento) em array.enumerate () {

// index é indexação da matriz

// elemento é elemento da matriz

}

Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.