Lua函数的多个返回值

来源：互联网发布：淘宝网红模特编辑：程序博客网时间：2024/05/06 16:59

Lua中的函数的一个很特殊也很有用的性质，即可以有多个返回值。包括一些内置的函数就是这样。比如string.find函数，在给定的字符串中查找一个pattern，如果有匹配的部分，则返回对应的头、尾的两个索引值；如果不存在匹配，则返回nil。当然，用户定义的函数也可以有多个返回值，通过return关键字返回多个结果即可。比如一个返回数组中最大值及其索引的函数：

[plain] view plaincopyprint?

function max(a)
local m = a[1];
local mIndex = 1;
for i,v in ipairs(a) do
if v>m then
m = v;
mIndex = i;
end
end
return m,mIndex;
end

function max(a)local m = a[1];local mIndex = 1;for i,v in ipairs(a) doif v>m thenm = v;mIndex = i;endendreturn m,mIndex;end

通常情况下，Lua会根据函数调用的不同情况自动调整返回值的有效个数。主要有以下几种情况：

1. 如果函数调用作为一个单独的语句，比如max({1,2,3})，则忽略所有的返回值；

2. 如果调用处在一串的表达式中：

1)如果处于表达式串的首位或中间，则只有第一个返回值被返回；

2)如果处于最后一个表达式，则所有的返回值都有效；

拿下面三个函数作为例子：

[plain] view plaincopyprint?

function f0() end;
function f1() return "a" end
function f2() return "a","b" end

function f0() end;function f1() return "a" endfunction f2() return "a","b" end

在多重赋值中，如果函数调用作为仅有的或最后一个表达式，则返回尽可能多的值来匹配左边的变量，如果多出，自动忽略；如果不够，赋为nil。比如，

[plain] view plaincopyprint?

x = f1() --x="a","b"被忽略
x,y = f1() --x="a",y="b"
x,y,z = f2() --x="a",y="b",z=nil
x,y,z = 10,f2() -- x=10, y="a", z="b"
x,y,z = f2(),10 -- x="a", y=10, z=nil, f2只返回一个值

x = f1()      --x="a","b"被忽略x,y = f1()  --x="a",y="b"x,y,z = f2() --x="a",y="b",z=nilx,y,z = 10,f2() -- x=10, y="a", z="b"x,y,z = f2(),10-- x="a", y=10, z=nil, f2只返回一个值

同理，作为另一个函数的参数时，规则一样：

[plain] view plaincopyprint?

print(f2()) --> a b
print(f2(), 1) --> a 1

print(f2()) --> a bprint(f2(), 1) --> a 1

还有用来构造数组时：

[plain] view plaincopyprint?

t = {f2()} -- t = {"a", "b"}
t = {f0(), f2(), 4} -- t[1] = nil, t[2] = "a", t[3] = 4

t = {f2()} -- t = {"a", "b"}t = {f0(), f2(), 4} -- t[1] = nil, t[2] = "a", t[3] = 4

最后，当函数返回值作为另一个函数的返回值被返回时，则全部被返回。如果想强制只返回一个值，则可以用括号 "()" 把表达式括起来即可。比如：

[plain] view plaincopyprint?

function foo (i)
if i == 0 then return foo0()
elseif i == 1 then return foo1()
elseif i == 2 then return foo2()
end
end

function foo (i)if i == 0 then return foo0()elseif i == 1 then return foo1()elseif i == 2 then return foo2()endend

[plain] view plaincopyprint?

print(foo(1)) --> a
print(foo(2)) --> a b
print(foo(0)) -- (无返回值)
print(foo(3)) -- (无返回值)

print(foo(1)) --> aprint(foo(2)) --> a bprint(foo(0)) -- (无返回值)print(foo(3)) -- (无返回值)

如果想强制只返回一个值，则可以用括号 "()" 把表达式括起来即可：

[plain] view plaincopyprint?

print((foo0())) --> nil
print((foo1())) --> a
print((foo2())) --> a

print((foo0())) --> nilprint((foo1())) --> aprint((foo2())) --> a

（见《Programming in Lua, 2rd》）