Golang处理CSV文件

1. 引言

CSV（Comma Separated Values）是一种常见的文件格式，在数据分析和数据交换中被广泛使用。CSV文件由逗号分隔的字段组成，每行表示一个记录。

Golang作为一种强大的静态类型语言，在处理CSV文件时提供了丰富的库和功能。本文将介绍如何使用Golang处理CSV文件，包括读取、写入和操作CSV文件的常用操作。

2. 读取CSV文件

首先，我们需要从CSV文件中读取数据。Golang提供了encoding/csv包来实现读取CSV文件的功能。以下是示例代码：

package main

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

func readCSVFile(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() {
    filename := "data.csv"

    records, err := readCSVFile(filename)
    if err != nil {
        fmt.Println("Failed to read CSV file:", err)
        return
    }

    for _, record := range records {
        for _, value := range record {
            fmt.Println(value)
        }
    }
}

在上述示例中，我们首先定义了一个readCSVFile函数，该函数接受一个文件名作为参数，并返回一个包含CSV文件内容的二维字符串切片。在函数内部，我们先打开文件，然后使用csv.NewReader创建一个新的读取器。然后，我们调用reader.ReadAll来读取文件中的所有记录，并将结果存储在records中。最后，我们返回records。

在main函数中，我们调用readCSVFile函数读取CSV文件，并使用嵌套的循环遍历并打印所有记录中的值。

3. 写入CSV文件

除了读取CSV文件外，我们还可以写入CSV文件。Golang的encoding/csv包提供了csv.Writer类型来实现写入CSV文件的功能。以下是示例代码：

package main

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

func writeCSVFile(filename string, data [][]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 data {
        err := writer.Write(record)
        if err != nil {
            return err
        }
    }

    return nil
}

func main() {
    filename := "output.csv"

    data := [][]string{
        {"Name", "Age", "Country"},
        {"Alice", "25", "USA"},
        {"Bob", "30", "Canada"},
        {"Charlie", "35", "UK"},
    }

    err := writeCSVFile(filename, data)
    if err != nil {
        fmt.Println("Failed to write CSV file:", err)
        return
    }

    fmt.Println("CSV file written successfully.")
}

在上述示例中，我们定义了一个writeCSVFile函数，该函数接受一个文件名和包含要写入文件的数据的二维字符串切片作为参数，并将数据写入CSV文件中。在函数内部，我们首先创建一个文件，然后使用csv.NewWriter创建一个新的写入器。然后，我们使用循环遍历数据，并调用writer.Write将每个记录写入文件。最后，我们返回nil表示写入成功。

在main函数中，我们定义了一个包含要写入CSV文件的数据的二维字符串切片data。然后，我们调用writeCSVFile函数将数据写入output.csv文件中。如果写入过程中出现错误，我们将打印出错信息。如果写入成功，我们将打印提示信息。

4. CSV文件的常用操作

在实际的应用场景中，我们通常需要对CSV文件进行一些常见的操作，例如过滤数据、排序数据等。以下是一些常见的CSV文件操作示例：

4.1. 过滤数据

我们可以使用Golang的切片操作来过滤CSV文件中的数据。以下是一个示例代码，演示如何过滤年龄大于等于30的记录：

package main

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

func filterData(records [][]string, column int, predicate func(value string) bool) [][]string {
    filteredRecords := [][]string{}
    for _, record := range records {
        if predicate(record[column]) {
            filteredRecords = append(filteredRecords, record)
        }
    }
    return filteredRecords
}

func main() {
    filename := "data.csv"

    records, err := readCSVFile(filename)
    if err != nil {
        fmt.Println("Failed to read CSV file:", err)
        return
    }

    ageColumn := 1

    agePredicate := func(value string) bool {
        age, err := strconv.Atoi(value)
        if err != nil {
            return false
        }
        return age >= 30
    }

    filteredRecords := filterData(records, ageColumn, agePredicate)

    for _, record := range filteredRecords {
        for _, value := range record {
            fmt.Println(value)
        }
    }
}

在上述示例中，我们定义了一个filterData函数，该函数接受一个二维字符串切片、一个表示要筛选的列的索引，以及一个函数作为参数。函数用于判断给定值是否满足筛选条件。filterData函数遍历所有记录，并根据筛选条件判断是否将记录添加到结果中。

在main函数中，我们从data.csv文件中读取CSV数据，并定义了一个表示年龄列的变量ageColumn。然后，我们定义了一个筛选条件函数agePredicate，用于判断年龄是否大于等于30。最后，我们调用filterData函数将满足筛选条件的记录打印出来。

4.2. 排序数据

我们可以使用Golang的sort.Slice函数来对CSV文件中的数据进行排序。以下是一个示例代码，演示如何按照年龄列对记录进行升序排序：

package main

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

func sortData(records [][]string, column int) {
    sort.Slice(records[1:], func(i, j int) bool {
        age1, _ := strconv.Atoi(records[i+1][column])
        age2, _ := strconv.Atoi(records[j+1][column])
        return age1 < age2
    })
}

func main() {
    filename := "data.csv"

    records, err := readCSVFile(filename)
    if err != nil {
        fmt.Println("Failed to read CSV file:", err)
        return
    }

    ageColumn := 1

    sortData(records, ageColumn)

    for _, record := range records {
        for _, value := range record {
            fmt.Println(value)
        }
    }
}

在上述示例中，我们定义了一个sortData函数，该函数接受一个二维字符串切片和一个表示要排序的列的索引作为参数。函数使用sort.Slice函数对记录进行排序，排序基于给定列的值。