Me gustaría extender Array en Swift para devolver un solo elemento en cada matriz, o columna, para una matriz 2D. Hasta ahora tengo:

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

Creo que necesito especificar de alguna manera una matriz 2D despuéswhere, pero no he podido encontrar la forma correcta de hacerlo.

¿Cuál es la sintaxis correcta para especificar una matriz 2D después dewhere?

También tengo curiosidad por saber si hay una buena documentación sobre cómo especificar para qué está disponible despuéswhere en una extensión vive. No pude encontrar eso enDocumentación de la extensión Swift de Apple

Gracias por adelantado.