Swift 中的 drop(where: {}) remove(where: {}) 函数？

Question

在 Swift 标准库中是否有一个函数作用于集合，接受一个谓词并返回从该集合中删除的值？

目前，我必须分两步实现它：

guard let idx = allAnnotations.index(where: {$0 is MKUserLocation}) else {return}
let userLocation = allAnnotations.remove(at: idx) as! MKUserLocation

但我猜，存在类似的功能。

目标

我有以下数组：

[Type1, Type1, Type1, Type1, Type1, Type1, Type2]

Type2 可能存在也可能不存在于数组中。除了这两种之外，没有其他类型。

我需要把它分成两个元素：

[Type1, Type1, Type1, Type1, Type1, Type1]

和

Type2?

这就是我要找的功能。

Answer 1

Swift 5 有removeAll(where)

@inlinable public mutating func removeAll(where shouldBeRemoved: (Element) throws -> Bool) rethrows

你可以这样使用它-

var array = [1,2,3]
array.removeAll(where: { $0 == 2})
print(array) // [1,3]

Apple Docs

Answer 2

这是一个返回被删除元素数组的扩展：

extension Array {
    mutating func dropWhere(_ isIncluded: (Element) throws -> Bool) -> [Element] {
        do {
            let reverseArray = try filter { try isIncluded($0) }
            self = try filter { try !isIncluded($0) }

            return reverseArray
        } catch {
            return []
        }
    }
}

就像调用 filter 一样调用它。

var array = [1, 2, 3]
let array2 = array.dropWhere { $0 > 2 }
print(array) //[1, 2]
print(array2) //[3]

Answer 3

我完成的代码：

extension Array {
  mutating func popFirstElementWith(_ predicate:(Element)->(Bool)) -> Element? {
    for (idx, element) in enumerated() {
      if predicate(element) {
        remove(at: idx)
        return element
      }
    }
    return nil
  }
}

Answer 4

你可以使用Split，条件是对你的数组进行切片

    var typeList =  [Type1, Type1, Type1, Type1, Type1, Type1, Type2]

    var slicedArray:[ArraySlice] = typeList.split { (value) -> Bool in
                return value == YourCondition
            }
print(slicedArray)

Answer 5

这个扩展接受一个谓词，如果它与谓词匹配，则返回元素。此外，从数组中删除元素。

extension Array {
    mutating func popFirstElementWith(_ predicate:(Element)->(Bool)) -> Element? {
        var firstElement:Element?
        self = self.compactMap({ (element:Element) -> Element? in
            guard predicate(element) == true else {
                return element
            }
            if firstElement == nil {
                firstElement = element
            }
            return nil
        })
        return firstElement
    }
}

测试：

var array:[Any] = ["a", "b", 1, "c"]
let element = array.popFirstElementWith { (element) -> (Bool) in
    return element is Int
}
print(element!)
print(array)

输出：

1
["a", "b", "c"]