Lua语言学习之迭代器

来源：互联网发布：李绳武知乎编辑：程序博客网时间：2024/05/20 14:42

1. 迭代器

迭代器（Iterator）一般用于泛型 for。所谓迭代器就是一种可以遍历（iterate over）一个

集合中所有元素的机制。在 Lua 中，通常将迭代器表示为函数。每调用一次函数，即返回

集合中的“下一个”元素。

例如：

function values ( t ) local i = 0return function ( ) i = i + 1; return t [ i ] endend

在 while 循环中使用该迭代器：

t = { 20,30,12 }iter = values ( t )while true dolocal element = iter ( )if element == nil thenbreak     end    print( element )end

使用泛型 for 则更为简单：

for element in values( t ) doprint ( element )end

2. 泛型 for 的语义

泛型 for 在循环过程内部保存了迭代器函数。实际上它保存着三个值：一个迭代器函数、一个

恒定状态（invariant state）和一个控制变量（control variable）。

泛型 for 的语法：

for < var-list > in < exp-list > do

<body>

end

< var-list > 是一个或多个变量名的列表，以逗号分隔；

< exp-list > 是一个或多个表达式的列表，以逗号分隔。

变量列表的第一个元素称为“控制变量”，当它为 nil 时循环结束。

for 做的第一件事就是对 in 后面的表达式求值。这些表达式应该返回三个值供 for 保存：

迭代器函数、恒定状态和控制变量的初值。在初始化步骤之后，for 会以恒定状态和控制

变量来调用迭代器函数。然后，for 将迭代器函数的返回值赋予变量列表中的变量。

如果第一个返回值为 nil，那么循环终止。否则，for 执行它的循环体，随后再次调用迭代器函数，

并重复这个过程。也就是等价于：

do local _f, _s, _var = <exp-list>while true dolocal var_1, ... , var_n = _f ( _s, _var )_var = var_1if _var == nil then break end<block>endend

2. 无状态的迭代器

所谓无状态的迭代器，如其名，就是一种自身不保存任何状态的迭代器。典型的例子就是

ipairs，它可以用来迭代一个数组的所有元素。

local function iter ( a, i )i = i + 1local v = a [ i ]if v thenreturn i, vendend

function ipairs ( a )return iter, a, 0end

函数 pairs 和 ipairs 类似，也是用于遍历一个 table 中的所有元素。不同的是，它的迭代器函数

是 Lua 中的一个基本函数 next 。

0 0