如何在Lua中高效优雅地对数组元素执行数值运算？

Lua本身没有像R那样原生支持向量式的批量运算，但你可以自己封装一个通用的map函数，来实现给表中每个元素应用指定操作的需求，用起来就会和R的写法一样简洁。

通用map函数实现 #

function map(tbl, fn)
    local new_tbl = {}
    -- 针对连续整数索引的数组，用ipairs比pairs效率更高
    for idx, val in ipairs(tbl) do
        new_tbl[idx] = fn(val)
    end
    return new_tbl
end

使用示例 #

针对你给出的数组：

local a = {1, 2, 10, 15}
-- 每个元素除以3，生成新数组b
local b = map(a, function(x) return x / 3 end)

这个map函数还能适配其他元素操作，比如：

-- 每个元素乘以2
local c = map(a, function(x) return x * 2 end)
-- 每个元素取平方
local d = map(a, function(x) return x ^ 2 end)

关于效率的说明 #

如果只是单次处理，你原本的循环写法其实已经足够高效——Lua的循环本身开销极低，没必要特意封装函数。但如果需要多次进行这类批量元素操作，封装map函数能极大提升代码的可读性和复用性。

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