【Lua 入门基础篇（三）】流程控制&函数&ipairs&pairs

一、循环控制

{}

1. if语句

a = 10
if a == 10 then
	print("a = 10")
end

2. if…else语句

a = 10
if a == 10 then
	print("a = 10")
else
	print("a != 10")
end

3. if…else嵌套语句

a = 10
if a == 10 then
	print("a = 10")
	if a % 5 == 0 then
		print("a % 5 = 0")
	end
else
	print("a != 10")
end

4. if…elseif…else语句

a = 5
if a % 2 == 0 then
    print("even")
elseif a % 3 == 0 then
    print("a % 3")
else
    print("odd")
end

---

二、函数

Lua 函数主要有两种用途：

• 完成指定的任务，这种情况下函数作为调用语句使用。
• 计算并返回值，这种情况下函数作为赋值语句的表达式使用。

1. 函数定义

Lua 编程语言函数定义格式如下：

optional_function_scope function function_name( argument1, argument2, argument3..., argumentn)
    function_body
    return result_params_comma_separated
end

• optional_function_scope: 该参数是可选的制定函数是全局函数还是局部函数，未设置该参数默认为全局函数，如果你需要设置函数为局部函数需要使用关键字 local。
• function_name: 指定函数名称。
• argument1, argument2, argument3…, argumentn: 函数参数，多个参数以逗号隔开，函数也可以不带参数。
• function_body: 函数体，函数中需要执行的代码语句块。
• result_params_comma_separated: 函数返回值，Lua语言函数可以返回多个值，每个值以逗号隔开。

---

2. 多返回值

Lua函数可以返回多个结果值，比如string.find，其返回匹配串"开始和结束的下标"（如果不存在匹配串返回nil）。

str = "www.baidu.com"
s, e = string.find(str, "baidu")
print(s, e)	 --> 5	9

• 返回最大值索引，最大值。

function max(tab)
    local index = 1
    local mx = tab[index]
    for i, v in ipairs(tab) do
        if v > mx then
            mx, index = v, i
        end
    end
    return index, mx
end

tab = {3, 5, 2, 1, 4}
print(max(tab))		--> 2	5

---

3. 可变参数

Lua 函数可以接受可变数目的参数，和 C 语言类似，在函数参数列表中使用三点 ... 表示函数有可变的参数。

function add(...)
    local sum = 0
    for i, v in ipairs{...} do	--> {...} 表示一个由所有变长参数构成的数组  
        sum = sum + v
    end
    return sum
end
print(add(1, 2, 3, 4))	--> 10

我们可以将可变参数赋值给一个变量。

function average(...)
    res = 0
    local tab = {...}
    for i, v in ipairs(tab) do
        res = res + v
    end
    return res / #tab
end
print(average(1, 2, 3, 4))	--> 2.5

我们也可以通过 select("#",...) 来获取可变参数的数量:

function count(...)
    print(select('#', ...))
end
count(1, 2, 3)

固定参数必须放在变长参数之前:

function printf(fmt, ...)
    print(fmt, ...)
end
printf("now: ", 2022, 07, 08)

通常在遍历变长参数的时候只需要使用 {…}，然而变长参数可能会包含一些 nil，那么就可以用 select 函数来访问变长参数了：select('#', …) 或者 select(n, …)

• select('#', …) ：返回可变参数的长度。
• select(n, …) ：返回从起点 n 开始到结束位置的所有参数列表。

function test(...)
    print(#{...})           --> 4
    print(select('#', ...)) --> 4
    a = select('2', ...)
    print(a)				--> 2
    print(select('4', ...)) --> 3
    for i, v in ipairs{...} do
        if v == nil then goto continue end
        print(v)
        ::continue::
    end
end

test(1, 2, nil, 3)          --> 1  2

发现两个有意思的点：

• select用返回值接住，和不接住打印结果不一样。

function test(...)
    print(select('2', ...)) --> 5	6	7
    a = select('2', ...)
    print(a)				--> 5
end

test(4, 5, 6, 7)

原因是： a 只接住了多个返回值的第一个。

• ipairs遇到nil会结束。

确实ipairs遍历到nil会结束，下述pairs和ipairs的区别。

---

三、pairs & ipairs 区别

1. 区别一

• ipairs遇到nil会停止。
• pairs会跳过nil，然后继续下去。

2. 区别二

• ipairs会跳过键值对，然后顺序输出table中的值，遇到nil则会停止。
• pairs会输出键值对，不过先顺序输出值，再乱序（键的哈希值）输出键值对。

local a = {'hello', 'world', a = 1, b = 2, 'Good', nil, 'Bye'}

print("=======ipairs========")
for i, v in ipairs(a) do
    print(v)
end

print("=======ipair=========")

for i, v in pairs(a) do
    print(i, v)
end

print("3========4=======5===")
print(a[3], a[4], a[5])

运行结果：

由上可以看出：

• table并不会给键值对一个索引值。
• ipairs只是按照索引值顺序，打印出了table中有索引的数据，没有索引值的不管。
• pairs是先按照索引值打印，打印完成后再按照键值对的键的哈希值打印它的值。

---

四、如何顺序输出table中的键值对？

在此就看一种方法：insert进一个新的数组，然后进行排序：

tab = {}
tab[3] = '2'
tab[2] = '1'
tab[5] = '3'
tab[8] = '4'

print("tab ---- ")
for k, v in pairs(tab) do
    print(k, v)
end

tab1 = {}
for k, v in pairs(tab) do
    local tmp = {k, v}
    table.insert(tab1, tmp)
end

print("insert ---- ")
for k, v in pairs(tab1) do
    print(k, v[1], v[2])
end

table.sort(tab1, function(a, b)
    -- return tonumber(a) < tonumber(b)
    return a[1] < b[1]
end)

print("sort =----")
for k, v in pairs(tab1) do
    print(k, v[1], v[2])
end

运行结果：

---