使用golang encoding/csv读写CSV文件

CSV（逗号分隔值）是一种常见的文本文件格式，用于存储表格数据。在大数据处理、数据分析和数据交互的场景中，CSV文件经常被使用。由于Go语言中内置了encoding/csv包，我们可以方便地读写CSV文件。

读取CSV文件

要从CSV文件中读取数据，我们首先需要导入encoding/csv包：

import "encoding/csv"

接下来，我们可以使用csv.NewReader函数创建一个CSV Reader对象：

csvFile, err := os.Open("data.csv")
if err != nil {
    log.Fatal(err)
}
defer csvFile.Close()

reader := csv.NewReader(csvFile)

通过reader对象，我们可以使用Read方法逐行读取文件中的数据：

for {
    record, err := reader.Read()
    if err == io.EOF {
        break
    }
    if err != nil {
        log.Fatal(err)
    }
    // 处理每一行的数据
}

在以上代码中，我们使用循环读取文件中的每一行数据，并将其存储在record变量中。如果遇到文件末尾（io.EOF），我们会退出循环。

写入CSV文件

要向CSV文件中写入数据，我们同样需要导入encoding/csv包，并创建一个CSV Writer对象：

writer := csv.NewWriter(csvFile)

然后，我们可以使用Write方法将数据写入文件：

data := []string{"John", "Doe", "[email protected]"}
err := writer.Write(data)
if err != nil {
    log.Fatal(err)
}

在以上代码中，我们使用Write方法将一个字符串切片data写入文件。如果需要写入多行数据，只需要循环调用Write方法即可。

自定义CSV字段分隔符

CSV文件默认使用逗号作为字段分隔符，但有时候我们可能需要使用其他字符作为分隔符。通过设置CSV Writer对象的Comma字段，我们可以更改分隔符：

writer.Comma = ';'

在以上代码中，我们将分隔符设置为分号。现在，当我们使用Write方法写入数据时，数据将以分号分隔。

处理包含特殊字符的CSV文件

有时候，CSV文件中的某些字段可能包含特殊字符（如逗号、引号等），这会导致解析和写入CSV文件时出现问题。为了处理这种情况，encoding/csv包提供了引号机制。

当一个field中包含分隔符或引号字符时，该field会被包裹在双引号中。另外，如果field中包含双引号字符，该字符将被转义为两个连续的双引号。

在读取CSV文件时，我们不需要做任何特殊处理，encoding/csv包会自动处理引号和转义字符。在写入CSV文件时，我们需要使用Write方法的相应选项启用引号机制：

writer := csv.NewWriter(csvFile)
writer.UseFields(true)      // 启用引号机制
writer.FieldsPerRecord = -1 // 不限制每行字段数

以上代码中，我们使用UseFields方法启用引号机制，并将FieldsPerRecord设置为-1，表示不限制每行字段数。

总结

通过使用encoding/csv包，我们可以方便地读写CSV文件。在读取CSV文件时，我们可以使用CSV Reader对象的Read方法逐行读取数据，并对每一行进行处理。在写入CSV文件时，我们可以使用CSV Writer对象的Write方法将数据写入文件中。同时，我们还可以自定义字段分隔符和处理包含特殊字符的CSV文件。