将两个可变参数函数的结果合并。

假设我有两个可变参数函数,就像这样:

function a(num)
  if num == 1 then
    return 1
  else
    return 1, 2
  end
end

function b(num)
  if num == 1 then
    return 1
  else
    return 1, 2
  end
end

然后我想要构建另一个函数,调用 ab 并返回 a 的所有结果,然后是 b 的所有结果。我想写出像这样的代码:

function c(num)
  return a(num), b(num)
end

但它只会返回 a 的第一个结果,然后是 b 的所有结果。我该怎么做?

点赞
用户646619
用户646619

你只能在表达式列表中返回最后一个函数的所有结果; 其他结果将被截断为一个结果。

因此,这个代码:

function f1()
    return 1
end

function f2()
    return 2, 3
end

print(f1(), f2())

期望输出 1 2 3,但这段代码

print(f2(), f1())

会输出 2 1,因为 f2() 被截断为一个结果。

为了解决这个问题,如果你事先知道结果数量,可以这样做:

local a, b = f1()
local c, d = f2()
return a, b, c, d

对于任意数量的结果,你可以这样做:

local t1 = {f1()}
local t2 = {f2()}
-- 将 t2 添加到 t1
return unpack(t1)
2014-01-09 23:11:25
用户2226988
用户2226988

当你将列表扩展到列表上下文中时,只使用第一个项目,或者如果扩展发生在列表上下文的末尾,则使用所有项目。 (抱歉,对于 Perl 可能的术语。)

您可以在表构造器 { a(num) } 中捕获所有列表项。由于列表在列表上下文中作为最后一项扩展,因此使用所有项目。

回到另一个方向,您可以使用 unpack 函数将表缩小成列表。但是,它使用表“长度”概念,该概念仅适用于连续序列数组。由于函数结果可能在任何地方包含 nil,因此必须通过计数来测量表中的项目数,并使用 pairs 函数迭代表。

local function a(num)
  if num == 1 then
    return 1
  else
    return nil, 2
  end
end

local function b(num)
  if num == 1 then
    return 1
  else
    return nil, 2
  end
end

.

local function c(num)
    local t = {}
    local n = 0
    local bOffset = 0
    for k, v in pairs({ a(num) }) do
        table.insert(t, k, v)
        if (k > n) then
            n = k
        end
        if (k > bOffset) then
            bOffset = k
        end
    end
    for k, v in pairs({ b(num) }) do
        table.insert(t, bOffset + k, v)
        if (bOffset + k > n) then
            n = bOffset + k
        end
    end

    return unpack(t, 1, n)
end
print(nil,2,nil,2)
print(c(0202))
2014-01-10 02:55:15