如何优化PHP中删除嵌套数组指定键的函数？

核心优化点 #

原有逻辑本身功能是可行的，主要可优化空间只有两处：

• 循环内重复执行array_flip产生不必要的性能开销
• 手动遍历拼装新数组的写法可以简化，提升可读性

优化后代码（最优版，兼容PHP 7.4+） #

function removeSpecifiedColumns(array $data, array $columns): array
{
    // 提前翻转待删除键数组，全局仅执行1次
    $columnsToRemove = array_flip($columns);
    // 用array_map替代手动foreach拼装数组，代码更简洁
    return array_map(fn($row) => array_diff_key($row, $columnsToRemove), $data);
}

低版本兼容写法（PHP 7.4以下） #

如果运行环境不支持箭头函数，可以替换为常规匿名函数：

function removeSpecifiedColumns(array $data, array $columns): array
{
    $columnsToRemove = array_flip($columns);
    return array_map(function($row) use ($columnsToRemove) {
        return array_diff_key($row, $columnsToRemove);
    }, $data);
}

超大数组优化方案 #

如果需要处理十万级以上的大数组，可以使用引用遍历减少内存占用：

function removeSpecifiedColumns(array &$data, array $columns): array
{
    $columnsToRemove = array_flip($columns);
    foreach ($data as &$row) {
        $row = array_diff_key($row, $columnsToRemove);
    }
    // 主动销毁引用变量，避免后续操作产生非预期问题
    unset($row);
    return $data;
}

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