lua中的Python itertools产品功能

Question

我正在寻找一种方法来从 itetools 库中实现类似 python 的组合函数 product，在 lua 中使用 repeat 参数。

我找到了这个库 https://github.com/aperezdc/lua-itertools/blob/master/itertools.lua 但不确定这是否对我有用，因为我在那里找不到 product 函数

实现这种功能的算法是什么？

Answer 1

你可以使用这个功能：

function ProductRepeat(tab, repeatCount)
    local result = { {} }

    for _ = 1, repeatCount do
        local newResult = { }

        for _, r in ipairs(result) do
            for _, v in ipairs(tab) do
                table.insert(newResult, { v, unpack(r) })
            end
        end

        result = newResult
    end

    return result
end