Golang使用CSV读取、导出文件

是个懒人 · · 20288 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

导出csv文件

func WriteCsv() { //创建文件 f, err := os.Create("test.csv") if err != nil { panic(err) } defer f.Close() // 写入UTF-8 BOM f.WriteString("\xEF\xBB\xBF") //创建一个新的写入文件流 w := csv.NewWriter(f) data := [][]string{ {"1", "刘备", "23"}, {"2", "张飞", "23"}, {"3", "关羽", "23"}, {"4", "赵云", "23"}, {"5", "黄忠", "23"}, {"6", "马超", "23"}, } //写入数据 w.WriteAll(data) w.Flush() }

image.png

读取csv

func ReadCsv() { //准备读取文件 fileName := "test.csv" fs, err := os.Open(fileName) if err != nil { log.Fatalf("can not open the file, err is %+v", err) } defer fs.Close() r := csv.NewReader(fs) //针对大文件，一行一行的读取文件 for { row, err := r.Read() if err != nil && err != io.EOF { log.Fatalf("can not read, err is %+v", err) } if err == io.EOF { break } fmt.Println(row) } fmt.Println("\n---------------------------\n") // 针对小文件，也可以一次性读取所有的文件。注意，r要重新赋值，因为readall是读取剩下的 fs1, _ := os.Open(fileName) r1 := csv.NewReader(fs1) content, err := r1.ReadAll() if err != nil { log.Fatalf("can not readall, err is %+v", err) } for _, row := range content { fmt.Println(row) } }