在 Go 中加载 CSV 文件

开发运维 2022-10-11 共饮一杯手机阅读

这是在 Go 中逐行打开 CSV 文件的一种非常简单的方法。假设我们有一个pops.csv看起来像这样的 CSV 文件

城市	人口
拉各斯	21000000
东京	37400000
德里	28500000
里约热内卢	13300000

要逐行读取文件，

package main

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

type CityPopulation struct {
	City  string
	Population string
}

func check(e error) {
	if e != nil {
		panic(e)
	}

}

func main() {
	fmt.Println("This is the entry point!")

	file, err := os.Open("pops.csv")
	check(err)
	println("Log: Opened CSV file")

	reader := csv.NewReader(file)

	var pops []CityPopulation

	for {
		line, err := reader.Read()

		if err == io.EOF {
			break
		}
		check(err)

		pops = append(pops, CityPopulation{
			City:  line[0],
			Population: line[1],
		})
	}
	for i := range pops {
        fmt.Printf("The city of %s is populated with %s people.\n" , pops[i].City, pops[i].Population)
	}
}

我们创建了一个结构CityPopulation来保存 CSV 的值。然后我们创建了一个函数check(e error)来检查错误。该函数仅检查文件中的错误和恐慌。

在 main 函数中，我们使用os.Open("pops.csv")打开 CSV 文件并加载它。然后我们用它csv.NewReader()来创建一个阅读器。然后我们使用 for 循环逐行读取文件。请注意io.EOF，当它到达 CSV 文件的末尾时，它由 reader.Read() 发出。然后我们将该行的内容加载到pops之前创建的切片中。