arrays - 如何使用 Swift 从矩阵中获取子矩阵

Question

赞Jama Java中的package，有一个方法:

有没有类似功能的包？我已经搜索了一段时间，但 Surge 似乎没有帮助。我想知道我是否需要为此编写一个函数。

Answer 1

这可能是您正在寻找的:

func getSubmatrix(_ matrix: [[Double]], i0: Int, i1: Int, j0: Int, j1: Int) -> [[Double]] {
    var result = [[Double]]()

    for row in Array(matrix[i0...i1]) {
         result.append(Array(row[j0...j1]))
    }

    return result
}

你应该可以像这样使用它:

print(getSubmatrix([[1.3, 4.3, 5.2], [4.2, 1.6, 4.5], [6.8, 8.9, 10.1]], i0: 0, i1: 1, j0: 0, j1: 1))

这是目前的功能风格。如果您有/创建了一个 Matrix 对象，您可以将它作为实例方法添加到它并使用 self。