lua脚本：读csv文件中第i行第j个字段数据

最新推荐文章于 2024-08-24 08:59:13 发布

1055守门人

最新推荐文章于 2024-08-24 08:59:13 发布

阅读量423

点赞数 1

文章标签： lua 开发语言

本文链接：https://blog.youkuaiyun.com/weixin_49617318/article/details/137956231

版权

本文介绍了如何使用Lua编写read_csv_to_table和get_field_from_table函数，实现读取CSV文件并将数据存储在二维table中，同时展示了如何从table中按行和列检索字段，但提到了对于复杂格式和编码问题的处理局限性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

function read_csv_to_table(filename)
    local file = io.open(filename, "r")
    if not file then
        return nil, "无法打开文件"
    end

    local data = {}
    for line in file:lines() do
        local fields = {}
        for field in line:gmatch('([^,]+)') do
            table.insert(fields, field)
        end
        table.insert(data, fields)
    end

    file:close()
    return data
end

function get_field_from_table(data, i, j)
    if data and i > 0 and j > 0 and i <= #data and j <= #data[i] then
        return data[i][j]
    else
        return nil, "无效的索引"
    end
end


local filename = "1000_x01.csv"     -- 自己的文件
local i = 2  -- 读取第2行
local j = 3  -- 读取第3个字段

local data, err = read_csv_to_table(filename)
if not data then
    print("读取CSV文件时发生错误: " .. err)
else
    local field, err = get_field_from_table(data, 1, j)
    if field then
        print("第" .. i .. "行第" .. j .. "个字段的值是: " .. field)
    else
        print("获取字段时发生错误: " .. err)
    end
end

在这个脚本中，read_csv_to_table函数读取CSV文件，并将其内容存储在一个二维table中，每一行是一个包含字段的table。然后，get_field_from_table函数从这个table中检索特定行和列的字段。

另外要注意一件事，这个脚本假设CSV文件中的字段是由逗号分隔的，并且没有考虑引号或特殊字符。如果你的CSV文件有更复杂的格式（例如，包含引号或换行符的字段），就可能需要使用更复杂的解析器来正确读取文件。

此外，这个脚本也没有处理可能出现的编码问题。如果你的CSV文件使用的是非UTF-8编码，可能需要在读取文件之前将其转换为UTF-8编码，或者相应地调整Lua的文件读取代码。