无涯教程-Lua - Iterators(迭代器)

91 阅读2分钟

迭代器是一种构造,使您可以遍历所谓的集合或集合的元素。在Lua中,这些集合通常引用表,这些表用于创建各种数据结构(如数组)。

通用迭代器

通用的 for 迭代器提供集合中每个元素的键值对。下面给出一个简单的示例。

array={"Lua", "Tutorial"}

for key,value in ipairs(array) do print(key, value) end

当无涯教程运行上面的代码时,将获得以下输出-

1  Lua
2  Tutorial

上面的示例使用了Lua提供的默认 ipairs 迭代器函数。

在Lua中,使用函数来表示迭代器。基于这些迭代器函数的状态维护,有两种主要类型-

  • 无状态迭代器
  • 有状态迭代器

无状态迭代器

通过名称本身,可以理解这种类型的迭代器函数不保留任何状态。

现在让无涯教程看一个使用简单的函数创建自己的迭代器的示例,该函数打印 n 的平方。

function square(iteratorMaxCount,currentNumber)

if currentNumber<iteratorMaxCount then currentNumber=currentNumber+1 return currentNumber, currentNumber*currentNumber end

end

for i,n in square,3,0 do print(i,n) end

当运行上面的程序时,将得到以下输出。

1	1
2	4
3	9

可以稍微修改上面的代码,以模仿迭代器的 ipairs 函数的工作方式。如下所示。

function square(iteratorMaxCount,currentNumber)

if currentNumber<iteratorMaxCount then currentNumber=currentNumber+1 return currentNumber, currentNumber*currentNumber end

end

function squares(iteratorMaxCount) return square,iteratorMaxCount,0 end

for i,n in squares(3) do print(i,n) end

当运行上面的程序时,将得到以下输出。

1	1
2	4
3	9

有状态迭代器

之前使用函数进行迭代的示例未保留状态。每次调用该函数时,它都会根据发送给该函数的第二个变量返回集合的下一个元素。为了保持当前元素的状态,使用了闭包,闭包在函数调用之间保留变量值。

现在,让无涯教程看一个创建自己的迭代器的示例,其中将使用闭包。

array={"Lua", "Tutorial"}

function elementIterator (collection)

local index=0 local count=#collection

-- The closure function is returned

return function () index=index + 1

  </span><span class="kwd">if</span><span class="pln"> index </span><span class="pun">&lt;=</span><span class="pln"> count
  </span><span class="kwd">then</span><span class="pln">
     </span><span class="pun">--</span><span class="pln"> </span><span class="kwd">return</span><span class="pln"> the current element </span><span class="kwd">of</span><span class="pln"> the iterator
     </span><span class="kwd">return</span><span class="pln"> collection</span><span class="pun">[</span><span class="pln">index</span><span class="pun">]</span><span class="pln">
  </span><span class="kwd">end</span><span class="pln">
	

end

end

for element in elementIterator(array) do print(element) end

当运行上面的程序时,将得到以下输出。

Lua
Tutorial

在上面的示例中,无涯教程可以看到elementIterator内部有另一个方法,该方法使用局部外部变量index和count每次调用函数时通过增加索引来返回集合中的每个元素。

参考链接

www.learnfk.com/lua/lua-ite…