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

news2025/1/18 16:29:33

迭代器是一种构造，使您可以遍历所谓的集合或集合的元素。在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
		
      if index <= count
      then
         -- return the current element of the iterator
         return collection[index]
      end
		
   end
	
end

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