将 Csv 文件转换为 Lua 表并将每行作为新表或函数()访问

目前,我的代码包含每个对象所需的数据的简单表,如下所示:

infantry = {class = "army", type = "human", power = 2}

cavalry = {class = "panzer", type = "motorized", power = 12}

battleship = {class = "navy", type = "motorized", power = 256}

我在各种函数中使用表名作为标识符,以便将它们的值逐个处理为一个函数,只需要调用该函数即可访问这些值。

现在我想将这些数据存储在电子表格(CSV 文件)中,看起来像这样:

名称    类型   种类     功率

步兵   军队   人类        2

骑兵   装甲   机动化   12

战舰   海军建立 机动化  256

电子表格将不超过 50 行,并且我希望能够在未来增加列。

我尝试了一些来自类似情况的方法,但由于缺乏技能,失败了无法从嵌套表中访问任何值。我认为这是因为我没有完全理解从 csv 文件读取每行到表后表的结构是如何的,因此无法打印任何值。

如果有一种方法可以从表格中获取 名称、类别、类型和功率 并像旧的简单表一样使用该行,我将非常感激您提供一个教育示例。另一种方法是从 CSV 文件中逐行声明与我的旧简单表完全相同的新表。我不知道是否可行。

使用 Lua 5.1

点赞
用户7396148
用户7396148

你可以将csv文件读入为字符串。在这里,我将使用多行字符串表示csv。

gmatch与模式[^\n]+将返回csv的每一行。 gmatch与模式[^,]+将返回给定行中每列的值。

如果添加了更多行或列,或者移动了列,只要第一行具有标题信息,我们仍然可靠地转换信息。

唯一不能移动的列是第一列“Name”列。如果移动它,将会改变用于将行存储到表中的键。

使用gmatch和2个模式[^,]+[^\n]+,可以将字符串分隔为csv的每一行和列。以下是代码的注释:

local csv = [[
Name,class,type,power
Infantry,army,human,2
Cavalry,panzer,motorized,12
Battleship,navy,motorized,256
]]

local items = {}                      -- 在这里存储我们的值
local headers = {}                    --
local first = true
for line in csv:gmatch("[^\n]+") do
  if first then                       -- 处理第一行并捕获标题信息
    local count = 1
    for header in line:gmatch("[^,]+") do
      headers[count] = header
      count = count + 1
    end
    first = false                     -- 让first为false以关闭标题块
  else
    local name
    local i = 2                       -- 我们从2开始,因为不会为标题增加增量
    for field in line:gmatch("[^,]+") do
      name = name or field            -- 检查我们是否知道行的名称
      if items[name] then             -- 如果该名称已经在items表中,则这是一个字段
        items[name][headers[i]] = field -- 使用给定名称在表中的标题分配我们的值。
        i = i + 1
      else                            -- 如果该名称不在表中,则为其创建一个新索引
        items[name] = {}
      end
    end
  end
end

以下是如何使用I/O库加载csv的示例:

-- 加载csv的示例。
path = "some\\path\\to\\file.csv"
local f = assert(io.open(path))
local csv = f:read("*all")
f:close()

或者,您可以使用io.lines(path),它将取代for循环部分中的csv:gmatch(“ [^ \ n] +”)。

以下是使用生成的表的示例:

-- 输出表格
print("items = {")
for name, item in pairs(items) do
  print("    " .. name .. " = { ")
  for field, value in pairs(item) do
    print("        " .. field .. " = ".. value .. ",")
  end
  print("    },")
end
print("}")

输出:

items = {
    Infantry = {
        type = human,
        class = army,
        power = 2,
    },
    Battleship = {
        type = motorized,
        class = navy,
        power = 256,
    },
    Cavalry = {
        type = motorized,
        class = panzer,
        power = 12,
    },
}
2019-06-13 21:36:08