Golang中读写CSV文件的全面指南

package main
 
import (
    "encoding/csv"
    "fmt"
    "io"
    "os"
)
 
// 将数据写入CSV文件
func writeCSV(filename string, records [][]string) error {
    file, err := os.Create(filename)
    if err != nil {
        return err
    }
    defer file.Close()
 
    writer := csv.NewWriter(file)
    defer writer.Flush()
 
    for _, record := range records {
        err := writer.Write(record)
        if err != nil {
            return err
        }
    }
 
    return nil
}
 
// 从CSV文件读取数据
func readCSV(filename string) ([][]string, error) {
    file, err := os.Open(filename)
    if err != nil {
        return nil, err
    }
    defer file.Close()
 
    reader := csv.NewReader(file)
 
    records, err := reader.ReadAll()
    if err != nil {
        return nil, err
    }
 
    return records, nil
}
 
func main() {
    // 示例数据
    records := [][]string{
        {"Name", "Age", "City"},
        {"Alice", "30", "New York"},
        {"Bob", "25", "Los Angeles"},
    }
 
    // 写入CSV文件
    filename := "example.csv"
    err := writeCSV(filename, records)
    if err != nil {
        fmt.Println("写入CSV文件时出错:", err)
        return
    }
    fmt.Println("CSV文件写入成功:", filename)
 
    // 读取CSV文件
    readRecords, err := readCSV(filename)
    if err != nil {
        fmt.Println("读取CSV文件时出错:", err)
        return
    }
 
    // 打印读取的数据
    for _, record := range readRecords {
        fmt.Println(record)
    }
}

这段代码展示了如何在Go语言中创建、写入和读取CSV文件。writeCSV函数接受文件名和记录切片，将数据写入指定的CSV文件。readCSV函数只接受文件名，读取文件内容并返回记录切片。在main函数中，我们创建了一些示例数据，写入了一个CSV文件，然后读取并打印了这个文件的内容。