Видалення з масиву під час перерахування в Swift?

Я хочу перерахувати масив у Swift та видалити певні елементи. Мені цікаво, чи безпечно це робити, а якщо ні, як я маю цього досягти.

В даний час я буду робити це:

for (index, aString: String) in enumerate(array) {
    //Some of the strings...
    array.removeAtIndex(index)
}

У Swift 2 це досить просто використовувати enumerateі reverse.

var a = [1,2,3,4,5,6]
for (i,num) in a.enumerate().reverse() {
    a.removeAtIndex(i)
}
print(a)

Ви можете розглянути filterспосіб:

var theStrings = ["foo", "bar", "zxy"]

// Filter only strings that begins with "b"
theStrings = theStrings.filter { $0.hasPrefix("b") }

Параметр filterпросто - це закриття, яке приймає екземпляр типу масиву (у цьому випадку String) і повертає a Bool. Коли результат є, trueвін зберігає елемент, інакше елемент фільтрується.

У Swift 3 та 4 це буде:

З цифрами, відповідно до відповіді Джонстона:

var a = [1,2,3,4,5,6]
for (i,num) in a.enumerated().reversed() {
   a.remove(at: i)
}
print(a)

З рядками як запитання OP:

var b = ["a", "b", "c", "d", "e", "f"]

for (i,str) in b.enumerated().reversed()
{
    if str == "c"
    {
        b.remove(at: i)
    }
}
print(b)

Однак зараз у Swift 4.2 або пізнішої версії є навіть кращий, швидший спосіб, який рекомендував Apple у WWDC2018:

var c = ["a", "b", "c", "d", "e", "f"]
c.removeAll(where: {$0 == "c"})
print(c)

Цей новий спосіб має ряд переваг:

1. Це швидше, ніж реалізації з filter.
2. Це позбавляє від необхідності реверсивних масивів.
3. Він видаляє елементи на місці, і, отже, оновлює вихідний масив замість виділення та повернення нового масиву.

Коли елемент із певним індексом буде видалено з масиву, позиція (та індекс) усіх наступних елементів буде змінена, оскільки вони зміщуються назад на одну позицію.

Тож найкращий спосіб - це навігація масивом у зворотному порядку - і в цьому випадку я пропоную використовувати традиційний цикл for:

for var index = array.count - 1; index >= 0; --index {
    if condition {
        array.removeAtIndex(index)
    }
}

Однак, на мою думку, найкращим підходом є використання filterметоду, як описано @perlfly у своїй відповіді.

Ні, не безпечно мутувати масиви під час енумації, ваш код вийде з ладу.

Якщо ви хочете видалити лише кілька об'єктів, ви можете скористатися цією filterфункцією.

Або створіть змінний масив для зберігання елементів, які потрібно видалити, а потім після перерахування видаліть ці елементи з оригіналу. Або створіть копію масиву (незмінний), перелічіть це та видаліть об’єкти (не за індексом) з оригіналу під час перерахування.

Традиційний цикл for можна замінити простим циклом while, що корисно, якщо вам також потрібно виконати деякі інші операції з кожним елементом перед видаленням.

var index = array.count-1
while index >= 0 {

     let element = array[index]
     //any operations on element
     array.remove(at: index)

     index -= 1
}

Я рекомендую встановити для нуля елементи під час перерахування, а після завершення видалити всі порожні елементи за допомогою методу масивів filter ().

Просто додамо, якщо у вас є кілька масивів, і кожен елемент в індексі N масиву A пов’язаний з індексом N масиву B, ви все ще можете використовувати метод, що повертає перерахований масив (як минулі відповіді). Але пам’ятайте, що при доступі до елементів інших масивів та їх видаленні немає необхідності їх змінювати.

Like so, (one can copy and paste this on Playground)

var a = ["a", "b", "c", "d"]
var b = [1, 2, 3, 4]
var c = ["!", "@", "#", "$"]

// remove c, 3, #

for (index, ch) in a.enumerated().reversed() {
    print("CH: \(ch). INDEX: \(index) | b: \(b[index]) | c: \(c[index])")
    if ch == "c" {
        a.remove(at: index)
        b.remove(at: index)
        c.remove(at: index)
    }
}

print("-----")
print(a) // ["a", "b", "d"]
print(b) // [1, 2, 4]
print(c) // ["!", "@", "$"]