如何在Swift中实现类似NumPy的pad功能为二维数组补位扩容

原生Swift直接实现（无依赖） #

你需求的是对二维数组的第1维（列维度）尾部补0，直接通过数组拼接即可实现，适配你的场景代码如下：

// 原始mfccs为形状[40, 91]的二维Float数组
let mfccs: [[Float]] = // 你的原始数据
let maxPadLen = 174
let padWidth = maxPadLen - mfccs[0].count

// 合法性校验，避免出现负数填充
guard padWidth > 0 else {
    // 若原始列数大于等于目标长度，可根据需求添加截断逻辑
    fatalError("原始数组列长度已超过目标长度，请检查参数")
}

// 逐行补0
let paddedMfccs = mfccs.map { row in
    row + [Float](repeating: 0, count: padWidth)
}

// 最终paddedMfccs形状为[40, 174]，总元素数6960

通用封装实现（类NumPy pad接口） #

如果需要更通用的填充功能，可以封装二维数组扩展，支持指定轴、前后填充长度、自定义填充值：

extension Array where Element == [Float] {
    /// 二维数组常量填充
    /// - Parameters:
    ///   - axis: 填充轴，0为行维度，1为列维度
    ///   - before: 轴前填充长度
    ///   - after: 轴后填充长度
    ///   - constantValue: 填充常量值，默认为0
    func pad(axis: Int, before: Int = 0, after: Int = 0, constantValue: Float = 0) -> [[Float]] {
        guard axis == 0 || axis == 1 else {
            fatalError("仅支持二维数组的0/1轴填充")
        }
        if axis == 0 {
            // 行维度填充
            let colCount = self.first?.count ?? 0
            let prefix = Array(repeating: Array(repeating: constantValue, count: colCount), count: before)
            let suffix = Array(repeating: Array(repeating: constantValue, count: colCount), count: after)
            return prefix + self + suffix
        } else {
            // 列维度填充
            return self.map { row in
                Array(repeating: constantValue, count: before) + row + Array(repeating: constantValue, count: after)
            }
        }
    }
}

调用示例：

let paddedMfccs = mfccs.pad(axis: 1, after: padWidth)

性能优化建议 #

如果是实时音频特征处理等对性能要求高的场景，可以使用苹果官方Accelerate框架的向量操作优化填充逻辑，相比普通数组拼接性能提升30%以上，适合大批量数据处理。

内容的提问来源于stack exchange，提问作者Beowulf