代码之家 › 专栏 › 技术社区 › mfaani

Swift有没有简单的方法来获取2d数组中的列?[副本]

higher-order-functions multidimensional-array algorithm swift arrays

mfaani · 技术社区 · 8 年前

我想在Swift中扩展数组,以在2D数组的每个数组或列中返回单个元素。到目前为止,我已经:

extension Array where // what goes here?
    func getColumn( column: Int ) -> [ Int ] {
        return self.map { $0[ column ] }
    }
}

我认为我需要在之后以某种方式指定2D数组 where ,但我一直无法找到正确的方法。

指定后的2D数组的正确语法是什么 哪里 ?

我还想知道是否有一个好的文档来说明如何在之后指定可用的内容延长寿命。我在网上找不到 Apple's Swift extension documentation

提前谢谢。

0 回复 | 直到 10 年前

Martin R 7 年前

您需要约束 Element 数组的类型。下标方法在 CollectionType 协议:

public protocol CollectionType : Indexable, SequenceType {
    // ...
    public subscript (position: Self.Index) -> Self.Generator.Element { get }
    // ...
}

因此,您可以为元素为集合的数组定义扩展方法:

extension Array where Element : CollectionType {
    func getColumn(column : Element.Index) -> [ Element.Generator.Element ] {
        return self.map { $0[ column ] }
    }
}

let a = [[1, 2, 3], [4, 5, 6]]
let c = a.getColumn(1)

print(c) // [2, 5]

extension Array where Element : CollectionType {
    subscript(column column : Element.Index) -> [ Element.Generator.Element ] {
        return map { $0[ column ] }
    }
}

let a = [["a", "b", "c"], [ "d", "e", "f" ]]
let c = a[column: 2]
print(c) // ["c", "f"]

更新 Swift 3:

extension Array where Element : Collection {
    func getColumn(column : Element.Index) -> [ Element.Iterator.Element ] {
        return self.map { $0[ column ] }
    }
}

或作为下标:

extension Array where Element : Collection {
    subscript(column column : Element.Index) -> [ Element.Iterator.Element ] {
        return map { $0[ column ] }
    }
}

推荐文章

monksy · 在scala中测试高阶函数

8 年前

Duncan C · 这是Swift 4编译器错误吗?

8 年前

Nemanja Beric · 为什么对向量调用过滤器不能从向量中删除元素?

8 年前

Thomas Cook · Kotlin高阶函数合成

8 年前

s1m0nw1 · 作用域函数apply/with/run/also/let:它们的名称来自哪里?

8 年前

Alonso · 高阶函数调用奇怪的行为

8 年前

pashadia · 从变量中对新对象调用方法

9 年前

morne · 如果内部有脂肪箭头功能

9 年前

Akash Sethi · 具有“Val”和“Def”的高阶函数`

9 年前

Riadiani · 从Javascript中的函数获取值

10 年前