在Swift中，获取数组中最后两项的最简洁方法是什么？

Question

有没有更简洁的方法来获取Swift中数组的最后两项？一般来说，我试图避免这种方法，因为它很容易与索引一起排除。 （在本例中使用Swift 1.2。）

// Swift -- slices are kind of a hassle?
let oneArray = ["uno"]
let twoArray = ["uno", "dos"]
let threeArray = ["uno", "dos", "tres"]

func getLastTwo(array: [String]) -> [String] {
    if array.count <= 1 {
        return array
    } else {
        let slice: ArraySlice<String> = array[array.endIndex-2..<array.endIndex]
        var lastTwo: Array<String> = Array(slice)
        return lastTwo
    }
}

getLastTwo(oneArray)   // ["uno"]
getLastTwo(twoArray)   // ["uno", "dos"]
getLastTwo(threeArray) // ["dos", "tres"]

我希望能有更接近Python的便利。

## Python -- very convenient slices
myList = ["uno", "dos", "tres"]
print myList[-2:] # ["dos", "tres"]

Answer 1

  

myList[-2:]

是的，我有一个增强请求，要求提供否定索引表示法，我建议您也提交一个。

但是，你不应该比自己更努力。内置的全局suffix功能完全符合您的要求：

let oneArray = ["uno"]
let twoArray = ["uno", "dos"]
let threeArray = ["uno", "dos", "tres"]

let arr1 = suffix(oneArray,2) // ["uno"]
let arr2 = suffix(twoArray,2) // ["uno", "dos"]
let arr3 = suffix(threeArray,2) // ["dos", "tres"]

结果是切片，但如果需要，可以将其强制转换为数组。

Answer 2

使用Swift 4，根据您的需要，您可以选择以下模式之一，以便从数组的最后两个元素中获取一个新数组。

＃1。使用Collection协议的suffix(_:)方法

使用Swift 4，符合Collection协议的对象具有suffix(_:)方法。 suffix(_:)方法具有以下声明：

func suffix(_ maxLength: Int) -> Self.SubSequence

  

返回一个子序列，直到给定的最大长度，包含集合的最终元素。

用法：的

let array = [1, 2, 3, 4]
let arraySlice = array.suffix(2)
let newArray = Array(arraySlice)
print(newArray) // prints: [3, 4]

let array: [Int] = []
let arraySlice = array.suffix(2)
let newArray = Array(arraySlice)
print(newArray) // prints: []

＃2。使用Array下标

let array = [1, 2, 3, 4]
let range = array.index(array.endIndex, offsetBy: -2) ..< array.endIndex
//let range = array.index(array.endIndex, offsetBy: -2)... // also works
let arraySlice = array[range]
let newArray = Array(arraySlice)
print(newArray) // prints: [3, 4]

＃3。使用您自己的Collection扩展方法

extension Collection {
    func safeSuffix(_ maxLength: Self.IndexDistance) -> Array<Element> {
        guard let index = self.index(endIndex, offsetBy: -maxLength, limitedBy: startIndex) else {
            return Array(self[startIndex ..< endIndex])
        }
        return Array(self[index ..< endIndex])
    }
}

用法：

let array = [1, 2, 3, 4]
let newArray = array.safeSuffix(2)
print(newArray) // prints: [3, 4]

let array: [Int] = []
let newArray = array.safeSuffix(2)
print(newArray) // prints: []

Answer 3

在Swift 2中，您可以扩展CollectionType。这是一个例子（借用Rob Napier的答案）：

extension CollectionType {
    func last(count:Int) -> [Self.Generator.Element] {
        let selfCount = self.count as! Int
        if selfCount <= count - 1 {
            return Array(self)
        } else {
            return Array(self.reverse()[0...count - 1].reverse())
        }
    }
}

您可以在任何CollectionType上使用它。这是Array：

let array = ["uno", "dos", "tres"]
print(array.last(2)) // [dos, tres]

这是CharacterView：

let string = "looking"
print(string.characters.last(4)) // [k, i, n, g]

（请注意，我的示例在所有情况下都返回一个数组，而不是原始集合类型。）

Answer 4

更通用的答案......

let a1 = [1,2,3,4,5]
let a2 = ["1","2","3","4","5"]

func getLast<T>(array: [T], count: Int) -> [T] {
  if count >= array.count {
    return array
  }
  let first = array.count - count
  return Array(array[first..<first+count])
}

getLast(a1, count: 2) // [4, 5]
getLast(a2, count: 3) // ["3", "4", "5"]

Answer 5

  

Swift中数组的最后两项

编辑：首先检查myArray.count＆gt; = 2

let myArray2:Array? = myArray.count >= 2 ? [myArray[myArray.count-2], myArray[myArray.count-1]] : nil

这里它包含在一个函数中，它接受数组并返回一个包含最后两个的数组，否则如果传递的数组不包含至少两个项则返回nil。

func getLastTwo(myArray:[String]) -> [String]? {
        return myArray.count >= 2 ? [myArray[myArray.count-2], myArray[myArray.count-1]] : nil
    }

Answer 6

let items = [0, 2, 5, 3, 7, 6, 9, 10]
let count = items.count
let last2 = items[count - 2 ..< count] // [9, 10]

Answer 7

我怀疑它会让你更开心，但数学肯定更简单：

func getLastTwo(array: [String]) -> [String] {
    if array.count <= 1 {
        return array
    } else {
        return array.reverse()[0...1].reverse()
    }
}

请注意reverse()是懒惰的，所以这并不是特别昂贵。

Answer 8

在 swift 5 中，您可以使用后缀来获取最后一个对象，并使用前缀来获取第一个对象，这是一个示例：

let exampleArray = ["first text", "second text", "third text"]

let arr1 = exampleArray.suffix(2) // ["second text", "third text"]
let arr2 = exampleArray.prefix(2) // ["first text", "second text"]

结果是一个切片，但是您可以根据需要将其强制为Array。

Answer 9

Swift4解决方案：

let oneArray = ["uno"]
let twoArray = ["uno", "dos"]
let threeArray = ["uno", "dos", "tres"]

let arr1 = threeArray.suffix(from: threeArray.count-2) // ["dos", "tres"]

澄清Swift内置函数func suffix(from start: Int) -> ArraySlice<Element>的功能的其他示例是......

let arr2 = oneArray.suffix(from: 0) // ["uno"]
let arr3 = twoArray.suffix(from: 0) // ["uno", "dos"]
let arr4 = twoArray.suffix(from: 1) // ["dos"]
let arr5 = threeArray.suffix(from: 1) // ["dos", "tres"]
let arr6 = threeArray.suffix(from: 2) // ["tres"]