arrays - Swift 2.1 数组扩展objectsAtIndexes

Question

我正在尝试扩展一个数组以仅返回某些索引处的对象。 map 功能似乎是我的最佳选择。

extension Array {

    func objectsAtIndexes(indexes: [Int]) -> [Element?]? {
        let elements: [Element?] = indexes.map{ (idx) in
            if idx < self.count {
                return self[idx]
            }
            return nil
        }.filter { $0 != nil }
        return elements
    }
}

let arr = ["1", "2", "3", "4", "5"]
let idx = [1,3,5]

let x = arr.objectsAtIndexes(idx) //returns: [{Some "2"}, {Some "4"}]

当我尝试将结果转换为字符串数组时，出现 EXC_BAD_INSTRUCTION 错误:

let x = arr.objectsAtIndexes(idx) as? [String]

有什么方法可以返回非可选数组吗？我尝试过返回[Element]？来自扩展功能。 这会引发相同的错误。

Answer 1

以下代码使用 flatmap 函数解决了 Swift2.1 的问题。

extension Array {

    func objectsAtIndexes(indexes: [Int]) -> [Element] {
        let elements: [Element] = indexes.map{ (idx) in
            if idx < self.count {
                return self[idx]
            }
            return nil
            }.flatMap{ $0 }
        return elements
    }
}