使用Go语言处理CSV文件|极客笔记

使用Go语言处理CSV文件

CSV（Comma-Separated Values）是一种常见的数据格式，它以逗号作为字段的分隔符，通常用于存储表格数据。在本文中，我们将使用Go语言来处理CSV文件。首先，我们将学习如何读取和写入CSV文件，接着我们将讨论如何进行数据的处理和转换。

读取CSV文件

在Go语言中，可以使用encoding/csv包来读取和写入CSV文件。以下是一个简单的示例，演示了如何读取CSV文件并将其内容打印出来：

package main

import (
    "encoding/csv"
    "fmt"
    "os"
)

func main() {
    // 打开CSV文件
    file, err := os.Open("data.csv")
    if err != nil {
        fmt.Println("Error opening file:", err)
        return
    }
    defer file.Close()

    // 创建一个CSV Reader
    reader := csv.NewReader(file)

    // 读取CSV数据
    records, err := reader.ReadAll()
    if err != nil {
        fmt.Println("Error reading CSV data:", err)
        return
    }

    // 打印CSV数据
    for _, record := range records {
        fmt.Println(record)
    }
}

假设我们有一个名为data.csv的文件，其内容如下：

Name,Age,City
Alice,25,New York
Bob,30,San Francisco
Charlie,35,Los Angeles

运行以上代码，输出为：

[Name Age City]
[Alice 25 New York]
[Bob 30 San Francisco]
[Charlie 35 Los Angeles]

写入CSV文件

除了读取CSV文件外，我们还可以使用encoding/csv包来写入CSV文件。以下是一个示例代码，演示了如何将数据写入CSV文件：

package main

import (
    "encoding/csv"
    "fmt"
    "os"
)

func main() {
    // 创建一个CSV文件
    file, err := os.Create("output.csv")
    if err != nil {
        fmt.Println("Error creating file:", err)
        return
    }
    defer file.Close()

    // 创建一个CSV Writer
    writer := csv.NewWriter(file)
    defer writer.Flush()

    // 写入CSV数据
    records := [][]string{
        {"Name", "Age", "City"},
        {"Dan", "28", "Chicago"},
        {"Eve", "33", "Miami"},
        {"Frank", "40", "Seattle"},
    }

    for _, record := range records {
        err := writer.Write(record)
        if err != nil {
            fmt.Println("Error writing record to CSV:", err)
            return
        }
    }
}

运行以上代码后，将生成一个名为output.csv的文件，其内容为：

Name,Age,City
Dan,28,Chicago
Eve,33,Miami
Frank,40,Seattle

数据处理和转换

在处理CSV文件时，通常需要对数据进行一些处理和转换。例如，我们可能需要在CSV文件中过滤数据、对数据进行排序或者进行统计等操作。以下是一个示例代码，演示了如何读取CSV文件并对数据进行一些处理后写入新的CSV文件：

package main

import (
    "encoding/csv"
    "fmt"
    "os"
)

func main() {
    // 打开原始CSV文件
    file, err := os.Open("data.csv")
    if err != nil {
        fmt.Println("Error opening file:", err)
        return
    }
    defer file.Close()

    // 创建CSV Reader
    reader := csv.NewReader(file)

    // 读取CSV数据
    records, err := reader.ReadAll()
    if err != nil {
        fmt.Println("Error reading CSV data:", err)
        return
    }

    // 创建一个新的CSV文件
    outputFile, err := os.Create("filtered_data.csv")
    if err != nil {
        fmt.Println("Error creating file:", err)
        return
    }
    defer outputFile.Close()

    // 创建CSV Writer
    writer := csv.NewWriter(outputFile)
    defer writer.Flush()

    // 处理和转换数据
    for _, record := range records {
        // 过滤条件：年龄大于等于30岁
        age := record[1]
        if age >= "30" {
            err := writer.Write(record)
            if err != nil {
                fmt.Println("Error writing record to CSV:", err)
                return
            }
        }
    }
}

运行以上代码后，将生成一个名为filtered_data.csv的文件，其中只包含年龄大于等于30岁的数据。

通过以上示例，我们可以看到使用Go语言处理CSV文件是非常方便的。encoding/csv包提供了丰富的API，可以满足我们处理CSV文件的各种需求。