Lua迭代器以及各种源函数的实现

范型for

范型for的格式如下所示:

Lua 复制代码
for <var-list> in <exp-list> do
    <body> 
end

var-list指变量名列表,可以为多个,exp-list指表达式列表,通常情况下只有一个值。可以更具体地写为另一种形式:

Lua 复制代码
for var_1, ..., var_n in explist(s) do block end

例如代码块中常见的for k,v in pairs(tb) do end或者for k,v in ipairs(tb) do end,都是这种范型形式,如若转换成while形式则如下:

Lua 复制代码
local func = explist
local s
while true do
    local var_1, ..., var_n = func(s)
    if var_1 == nil then
        break
    end
    block
end

源函数iter、ipairs、pairs实现

可以随意写一个简单的迭代器循环:

Lua 复制代码
function iter(a)
    local i = 0
    return function()
        i = i + 1
        return a[i]
    end
end

for n in iter({1,2,3}) do
    print(n) -- 1  2  3
end

相当于是以下代码结构:

Lua 复制代码
local iterator = iter({1,2,3})
local n = iterator()
while n do
    print(n)
    n = iterator()
end

ipairs函数只不过是上面iter内部的return a[i]改为return i,a[i]罢了。

pairs函数则是用到了next方法:

Lua 复制代码
function pairsMySelf(a)
    local key,value = next(a)
    return function()
        local oldKey,oldValue = key,value
        if value ~= nil then
            key,value = next(a,key)
        end
        return oldKey,oldValue
    end
end

local tb = {
    ["a"] = 1,
    ["b"] = 1,
    ["c"] = 1,
}
for k,v in pairsMySelf(tb) do
    print(k,v)
end
--[[print:
            a       1
            b       1
            c       1
]]--
相关推荐
Touper.9 分钟前
JavaSE -- 泛型详细介绍
java·开发语言·算法
sun00770011 分钟前
std::forward作用
开发语言·c++·算法
Zhen (Evan) Wang1 小时前
(豆包)xgb.XGBRegressor 如何进行参数调优
开发语言·python
虾球xz1 小时前
CppCon 2018 学习:THE MOST VALUABLE VALUES
开发语言·c++·学习
阿蒙Amon2 小时前
C#扩展方法全解析:给现有类型插上翅膀的魔法
开发语言·c#
尘浮7282 小时前
60天python训练计划----day59
开发语言·python
Chef_Chen3 小时前
从0开始学习R语言--Day39--Spearman 秩相关
开发语言·学习·r语言
不学会Ⅳ3 小时前
Mac M芯片搭建jdk源码环境(jdk24)
java·开发语言·macos
好开心啊没烦恼4 小时前
Python 数据分析:计算,分组统计1,df.groupby()。听故事学知识点怎么这么容易?
开发语言·python·数据挖掘·数据分析·pandas
lljss20205 小时前
Python11中创建虚拟环境、安装 TensorFlow
开发语言·python·tensorflow