使用 Go 读取 CSV

CSV 简介

CSV（Comma-Separated Values ）格式是一种纯文本数据格式，用于以表格的形式存储数据。

它的语法规则是：

每一行代表一条记录（row）
每一列之间用英文逗号 "," 分隔（列字段）
每一行之间用换行符分隔。这里通常无需考虑 \n 和 \r\n 的区别，因为大多数 CSV 解析器（包括 Go 的 encoding/csv 包）都能自动处理这两种换行。
没有定义表头的概念（用户通常默认第一行是表头）
文件扩展名是 .csv

🌰 一个 CSV 文件的例子

csv 复制代码

name,age,city
Alice,30,Beijing
Bob,25,Shanghai

要读取一个 CSV，我们首先得创建一个专门读 CSV 的 Reader 。这个 Reader 的特点在于，它每次能读取 CSV 的一行。方法如下：

go 复制代码

func NewReader(r io.Reader) *Reader

通常情况下，CSV 都来自一个文件。所以就有了如下范式：

go 复制代码

file, err := os.Open("data.csv")
if err != nil {
	panic(err)
}
defer file.Close()

reader := csv.NewReader(file)

▶️ 然后，我们就可以调用这个 Reader 的 Read 方法了。就像我们刚才所说的，它每次返回一行 CSV 数据。

go 复制代码

func (r *Reader) Read() (record []string, err error)

如果没有更多数据了，该方法就会返回 nil, io.EOF 。

和很多 Reader 类似，这里也有 ReadAll 方法，不过它返回的不是 []byte，而是入乡随俗的 [][]string 。和其它 ReadAll 方法一样，这里读完后也不会返回 io.EOF 而是返回 nil 。

go 复制代码

func (r *Reader) ReadAll() (records [][]string, err error)

接下来，我们就直接能用 for range 来迭代它了。

🎈 注意：无论用哪个方法，读出来的数据都是字符串类型的。

📢 本文为原创文章，首发于掘金。

🚫 禁止任何形式的转载、摘编、改编，尤其是未经授权的商业使用。