Golang 随机读取 CSV 行|极客教程

Golang 随机读取 CSV 行

引言

CSV（Comma-Separated Values）是一种常用的数据存储格式，用于在不同系统之间以简单的文本形式传递表格数据。在很多情况下，我们需要从一个大规模的 CSV 文件中随机读取一部分行数据进行处理。本文将介绍如何使用 Golang 随机读取 CSV 文件的行。

读取 CSV 文件

在 Golang 中，我们可以使用 encoding/csv 包来读取 CSV 文件。首先，我们需要导入该包：

import "encoding/csv"

然后，使用 csv.NewReader 函数创建一个 CSV 读取器对象。该函数需要传入一个实现了 io.Reader 接口的文件对象：

file, _ := os.Open("data.csv")
defer file.Close()

reader := csv.NewReader(file)

接下来，我们可以使用 Read 方法来读取一行 CSV 数据：

record, _ := reader.Read()

返回的 record 变量是一个字符串切片，包含了一行 CSV 数据的各个字段。我们可以通过索引来访问其中的字段。

需要注意的是，在 Golang 中，索引从 0 开始。所以，如果我们想要获取第一个字段的值，应该使用 record[0]。

统计 CSV 行数

在开始随机读取 CSV 行之前，我们可能需要先统计一下 CSV 文件中的总行数。可以使用以下代码来实现：

file, _ := os.Open("data.csv")
defer file.Close()

reader := csv.NewReader(file)

lines := 0
for {
    _, err := reader.Read()
    if err == io.EOF {
        break
    }
    lines++
}

fmt.Println("Total lines:", lines)

上述代码通过循环读取直到 io.EOF 错误（表示文件结束）出现，每读取一行，行数加一。

随机读取 CSV 行

有了总行数之后，我们可以使用 rand 包来生成一个随机行号。然后，我们可以使用 Seek 方法将文件指针移动到该行的位置。接下来，我们可以读取该行数据进行处理。

下面是一个随机读取 CSV 行的示例代码：

file, _ := os.Open("data.csv")
defer file.Close()

reader := csv.NewReader(file)

// 统计总行数
lines := 0
for {
    _, err := reader.Read()
    if err == io.EOF {
        break
    }
    lines++
}
fmt.Println("Total lines:", lines)

// 生成随机行号
rand.Seed(time.Now().UnixNano())
randLine := rand.Intn(lines) + 1
fmt.Println("Random line:", randLine)

// 移动文件指针到随机行
_, err := file.Seek(0, io.SeekStart)
if err != nil {
    fmt.Println(err)
    return
}
for i := 1; i < randLine; i++ {
    _, err := reader.Read()
    if err != nil {
        fmt.Println(err)
        return
    }
}

// 读取随机行数据
record, err := reader.Read()
if err != nil {
    fmt.Println(err)
    return
}

// 处理随机行数据
fmt.Println("Random line data:", record)

运行上述代码，你将看到如下输出：

Total lines: 100
Random line: 42
Random line data: [42 Field1 Field2 Field3 ...]

说明代码成功地随机读取了 CSV 文件的一行数据，并进行了处理。

总结

通过使用 Golang 的 encoding/csv 包，我们可以方便地读取和处理 CSV 文件的行数据。本文介绍了如何统计 CSV 行数以及如何随机读取一行数据。

Golang 随机读取 CSV 行