Torch/lua中如何将两个数组合并?

我在Torch/lua中有以下的数组:

aaa = {1, 2, 3, 4}

bbb = {0, 0, 0, 1}

我想要使用一个函数将它们合并成一个唯一的数组,按照位置和数组的顺序排序(a1、b1、a2、b2、a3、b3,...)。

例如,aaabbb 的合并输出应该为:

merged = {1, 0, 2, 0, 3, 0, 4, 1}

我该怎么做?

谢谢

点赞
用户7504558
用户7504558

以下是索引表(数组)的示例:

local aaa = {1, 2, 3, 4}
local bbb = {0, 0, 0, 1}
local max= (#aaa>#bbb) and #aaa or #bbb
local merged = {}
for i=1,max do
   if aaa[i] then merged[#merged+1] = aaa[i] end
   if bbb[i] then merged[#merged+1] = bbb[i] end
end

for k,v in pairs(merged) do
 print(k,v)
end
2017-10-30 20:26:27