如何在Lua中将数组转换为2D矩阵?

时间:2016-01-31 13:52:12

标签: lua

我有以下数组。

arr = {3412323450, 8912745671, 3212367894}

我想将其转换为简单的二维矩阵。

mat = {
         {3, 4, 1, 2, 3, 2, 3, 4, 5, 0},
         {8, 9, 1, 2, 7, 4, 5, 6, 7, 1},
         {3, 2, 1, 2, 3, 6, 7, 8, 9, 4}
}

最初,我会迭代arr,将其转换为字符串,然后拆分字符串,遍历每个字符串char并将其转换回数字,相应地存储mat中的每一行和数字。这真的很难看。

  1. 是否有更传统的方法将数组转换为矩阵 Lua的?
  2. 是否有人们经常使用的luarock包将数组转换为矩阵?

2 个答案:

答案 0 :(得分:4)

就我个人而言,我认为转换为字符串并抓住所有数字远比替代品更加漂亮(大致除以10,或者您可以想到的任何其他精心设计的方法)。如果您将操作包装在函数中,则尤其如此,因此您的转换不会始终出现在整个代码中。

function Digits(n)
    local digits = {}
    for d in tostring(n):gmatch('%d') do
        digits[#digits+1] = tonumber(d)
    end
    return digits
end

function ArrayToMatrix(array)
    local matrix = {}
    for i,v in ipairs(array) do
        matrix[i] = Digits(v)
    end
    return matrix
end

答案 1 :(得分:2)

好的,这是我的尝试。

arr = {3412323450, 8912745671, 3212367894}

function arr2matrix(arr)
  local mat = {}
  for i, row in ipairs(arr) do
    mat[i] = {}
    local j = 0
    row_str = string.gsub(row, '%d', '%0 ')
    for c in string.gmatch(row_str, '%S') do
      j = j + 1
      mat[i][j] = tonumber(c)
    end
  end
  return mat
end

-- checking the result
m = arr2matrix(arr)
for i=1, #m do
  for j=1, #m[i] do
    io.write(m[i][j]..',')
  end
  io.write('\n')
end

运行以上命令:

3,4,1,2,3,2,3,4,5,0,
8,9,1,2,7,4,5,6,7,1,
3,2,1,2,3,6,7,8,9,4,