Swift：从数组中删除特定对象的更好方法？

Question

我正在努力提高我的代码效率，但我脑子里放了个屁。我编写的这段代码效果很好，并且完全符合我的需要：它检查一个数组并删除一个未知索引处的对象。但我觉得有一种更好、更有效的方式来编写它。我去了 Array.remove(at:) 但这需要一个已知的索引。我正在使用大 O 表示法，不知道如何使它更容易处理。有什么想法吗？

 // create a new object array
 var sort : [MyCustomObject] = []

//iterate through my object array  

        for i in objectArray{
            if i === objectToRemove{
            }
            else{
                sort.append(i)
            }
        }

     // set old array to sort, which no longer has the unwanted object 
        self.objectArray = sort

Answer 1

如果您使用 Xcode 10.0+ beta（Swift 4.2 或更高版本）进行编码，您可以使用新方法 removeAll(where:)

mutating func removeAll(where predicate: (Element) throws -> Bool) rethrows

讨论：使用此方法删除集合中的每个元素 符合特定标准的。 复杂度：O(n)，其中 n 是集合的长度。

这个例子去掉了所有的奇数 来自数字数组的值：

Swift 5.2 或更高版本

extension BinaryInteger {
    var isEven: Bool { isMultiple(of: 2) }
    var isOdd: Bool { !isMultiple(of: 2) }
}

var numbers = [5, 6, 7, 8, 9, 10, 11]
numbers.removeAll(where: \.isOdd) // numbers == [6, 8, 10]
numbers

---

在您的情况下，请确保 MyCustomObject 符合 Equatable

objectArray.removeAll(where: { $0 == objectToRemove })

或使用符合它的属性之一作为谓词（即id: Int）：

objectArray.removeAll(where: { $0.id == idToRemove })

---

注意：如果您不使用 Xcode 10.0+ beta (Swift 4.2)，您可以实现自己的 removeAll(where:) 方法，正如您在此 answer 中看到的那样。

---

实现removeFirst(where:) 和removeLast(where:) 以避免迭代整个集合，正如@vacawama 在 cmets 中提到的那样

斯威夫特 4.1

extension RangeReplaceableCollection  {
    @discardableResult
    mutating func removeFirst(where predicate: (Element) throws -> Bool) rethrows -> Element?  {
        guard let index = try index(where: predicate) else { return nil }
        return remove(at: index)
    }
}

---

extension RangeReplaceableCollection where Self: BidirectionalCollection {
    @discardableResult
    mutating func removeLast(where predicate: (Element) throws -> Bool) rethrows -> Element? {
        guard let index = try indices.reversed().first(where: {
            try predicate(self[$0])
        }) else { return nil }
        return remove(at: index)
    }
}

---

Swift 4.2 或更高版本（由 @Hamish 建议）

extension RangeReplaceableCollection {
    @discardableResult
    mutating func removeFirst(where predicate: (Element) throws -> Bool) rethrows -> Element? {
        guard let index = try firstIndex(where: predicate) else { return nil }
        return remove(at: index)
    }
}

---

extension RangeReplaceableCollection where Self: BidirectionalCollection {
    @discardableResult
    mutating func removeLast(where predicate: (Element) throws -> Bool) rethrows -> Element? {
        guard let index = try lastIndex(where: predicate) else { return nil }
        return remove(at: index)
    }
}

您还可以查看 post 以了解 remove(while:)、removeLast(while:) 和 dropLast(while:) 方法的实现。

Answer 2

您可以使用此泛型从数组中删除任何类型的对象

extension Array where Element: Equatable {
mutating func removeObject(object: Element)  {
    if let index = firstIndex(of: object) {
        remove(at: index)
    }
}}

Answer 3

使用firstIndex(where:)（在Swift 4.1 及更早版本中称为index(where:)）使用谓词{ $0 === objectToRemove } 在数组中搜索对象，然后在数组上调用remove(at:) 以删除它：

if let idx = objectArray.firstIndex(where: { $0 === objectToRemove }) {
    objectArray.remove(at: idx)
}

这允许您搜索您的对象，无论它是否为Equatable。

Answer 4

斯威夫特 4

使用enumerated()，您将获得索引：

var array = ["a","b","c"]
let objectToRemove = "b"
for (index, value) in array.enumerated() {
    if value == objectToRemove {
        array.remove(at: index)
    }
}
print(array) // Prints: ["a","c"]

编辑：

或者也许更好，使用过滤器：

var array = ["a","b","c"]
let objectToRemove = "b"
array = array.filter { $0 != objectToRemove } // Keeps only what is
                                              // different from the 
                                              // objectToRemove
print(newArray)  // Prints: ["a","c"]

Answer 5

你可以试试

if let ind = objectArray.index(of:objectToRemove) {
  objectArray.remove(at:ind)
}