Swift Array 实例方法 drop(at: Int)

Question

Array in Swift有几种排除元素的实例方法，如dropFirst()、dropLast()、drop(while:)等，那么drop(at:)呢？

注意：我会使用 remove(at:)，但我正在使用的数组是 let 常量。

Answer 1

注意：我会使用 remove(at:)，但我正在使用的数组是一个常量。

我不会让这阻止你：

extension Array {
    func drop(at:Int) -> [Element] {
        var temp = self
        temp.remove(at: at)
        return temp
    }
}
let numbers = [1, 2, 3, 4, 5]
print(numbers.drop(at: 2)) // [1, 2, 4, 5]

Answer 2

您可以扩展RangeReplaceableCollection 协议而不是Array 类型，这样您也可以在字符串上使用它：

extension RangeReplaceableCollection {
    func drop(at offset: Int) -> SubSequence {
        let index = self.index(startIndex, offsetBy: offset, limitedBy: endIndex) ?? endIndex
        let next = self.index(index, offsetBy: 1, limitedBy: endIndex) ?? endIndex
        return self[..<index] + self[next...]
    }
}

---

var str = "Hello, playground"
str.drop(at: 5)  // "Hello playground"

---

let numbers = [1, 2, 3, 4, 5]
print(numbers.drop(at: 2))  // "[1, 2, 4, 5]\n"

---

如果你也想在你的方法中接受 String.Index：

extension RangeReplaceableCollection {
    func drop(at index: Index) -> SubSequence {
        let index = self.index(startIndex, offsetBy: distance(from: startIndex, to: index), limitedBy: endIndex) ?? endIndex
        let next = self.index(index, offsetBy: 1, limitedBy: endIndex) ?? endIndex
        return self[..<index] + self[next...]
    }
}

---

var str = "Hello, playground"
str.drop(at: 0)               // "ello, playground"
str.drop(at: str.startIndex)  // "ello, playground"

Answer 3

我要添加到您的实现中的唯一内容是带有有用错误消息的保护语句：

extension Array {
    func drop(at index: Int) -> ArraySlice<Element> {
        guard indices.contains(index) else {
            fatalError("Index out of range")
        }
        return self[0..<index] + self[(index+1)..<endIndex]
    }

    func drop(range: CountableRange<Int>) -> ArraySlice<Element> {
        guard (indices.lowerBound, range.upperBound) <= (range.lowerBound, indices.upperBound) else {
            fatalError("Range is out of the indices bounds")
        }
        return self[0..<range.lowerBound] + self[range.upperBound..<endIndex]
    }
}

let a = [1,2,3]
a.drop(at: 3)          //Fatal error: Index out of range
let b = [0,1,2,3,4,5]
b.drop(range: 1..<5)   //[0, 5]

Answer 4

return self[0..<index] + self[index+1..<endIndex]

丑陋。为什么不使用您提供的工具？

extension Array {
    func drop(at:Int) -> ArraySlice<Element> {
        return prefix(upTo: at) + suffix(from: at+1)
    }
}
let arr = [1,2,3,4,5]
let slice = arr.drop(at:2) // [1,2,4,5]

编辑似乎苹果更希望你说（使用部分范围）

return self[..<at] + self[(at+1)...]

但我个人认为这更丑陋，毕竟我建议的所有方法都没有被弃用。

Answer 5

使用Swift extension 将drop(at:) 添加到Array 结构怎么样？

extension Array {
  func drop(at index: Int) -> ArraySlice<Element> {
    precondition(indices.contains(index), "Index out of range")
    return self[..<index] + self[(index+1)...]
  }
}

它返回原始数组的切片，没有指定索引处的元素。

let numbers = [1, 2, 3, 4, 5]
print(numbers.drop(at: 2))
// Prints "[1, 2, 4, 5]"

注意：您可能还想将drop(at:) 添加到ArraySlice。