Lua - 具有可变数量参数的迭代器

Question

我一直在尝试了解 Lua 迭代器的工作方式和行为方式。虽然闭包对我来说很清楚，但我在使用无状态迭代器时遇到了困难。

最简单的任务，例如重新实现 ipairs 或pairs，对我来说是相当清楚的。但是如果我想编写一个迭代器，它接收可变数量的表并遍历它们怎么办？ 例如，

for i in iter({1,2,3}, {'a', 'b', 'c'}, {100, 101}, {}) do
    print(i)
end
--I would expect it to write: 1, 2, 3, a, b, c, 100, 101

我一直试图在互联网上找到一些有用的信息，但我所看到的只是一遍又一遍地研究相同的简单示例。所以，欢迎任何提示和建议！

Answer 1

我认为您正在谈论的迭代器应该如下所示： 编辑：用一个包含两个索引的数组替换了 2 个索引的使用。

function iter_variable (a, indexArr)
  indexArr[2] = indexArr[2] + 1
  local v = a[indexArr[1]][indexArr[2]]
  if v then
    return indexArr, v
  else
    -- end of current internal DS, move on to the next internal DS
    indexArr[1] = indexArr[1] + 1
    indexArr[2] = 1
    local v = a[indexArr[1]][indexArr[2]]
    if v then
        return indexArr, v
    end
  end
end

function ipairs_variable (...)
  local a = {...}
  return iter_variable, a, {1,0}
end

for i, v in ipairs_variable({1,2,3}, {'a', 'b', 'c'}, {100, 101}, {}) do
  print(v)
end