08.8 读取CSV文件

Mastering_Go_ZH_CN

Search…

介绍

[0 前言]

9 并发-Goroutines,Channel和Pipeline

08.8 读取CSV文件

CSV文件是纯文本文件。在本节中，你将学习如何读取包含平面点的文本文件，这意味着每一行将包含一对坐标。此外，你还将使用一个名为Glot的外部Go库，它将帮助你创建从CSV文件中读取的点的图表。注意，Glot使用Gnuplot，这意味着你需要在Unix机器上安装Gnuplot才能使用Glot。
本章的程序是CSVplot.go，分为五部分。第一部分代码如下：
1
package main
2
​
3
import (
4
    "encoding/csv"
5
    "fmt"
6
    "github.com/Arafatk/glot"
7
    "os"
8
    "strconv"
9
)
Copied!
CSVplot.go第二部分代码如下：
1
func main() {
2
    if len(os.Args) != 2 {
3
        fmt.Println("Need a data file!")
4
        return
5
    }
6
​
7
    file := os.Args[1]
8
    _, err := os.Stat(file)
9
    if err != nil {
10
        fmt.Println("Cannot stat", file)
11
        return
12
    }
Copied!
在本部分中，你将看到一种使用强大的os.Stat()函数检查文件是否已经存在的技术。
CSVplot.go第三部分代码如下：
1
    f, err := os.Open(file)
2
    if err != nil {
3
        fmt.Println("Cannot open", file)
4
        fmt.Println(err)
5
        return
6
    }
7
    defer f.Close()
8
​
9
    reader := csv.NewReader(f)
10
    reader.FieldsPerRecord = -1
11
    allRecords, err := reader.ReadAll()
12
    if err != nil {
13
        fmt.Println(err)
14
        return
15
    }
Copied!
CSVplot.go第四部分代码如下：
1
    xP := []float64{}
2
    yP := []float64{}
3
​
4
    for _, rec := range allRecords {
5
        x, _ := strconv.ParseFloat(rec[0], 64)
6
        y, _ := strconv.ParseFloat(rec[1], 64)
7
        xP = append(xP, x)
8
        yP = append(yP, y)
9
    }
10
​
11
    points := [][]float64{}
12
    points = append(points, xP)
13
    points = append(points, yP)
14
    fmt.Println(points)
Copied!
此处，你将字符串转为数字，并追加到二维切片points中。
CSVplot.go最后一部分代码如下：
1
    dimensions := 2
2
    persist := true
3
    debug := false
4
    plot, _ := glot.NewPlot(dimensions, persist, debug)
5
​
6
    plot.SetTitle("Using Glot with CSV data")
7
    plot.SetXLabel("X-Axis")
8
    plot.SetYLabel("Y-Axis")
9
    style := "circle"
10
    plot.AddPointGroup("Circle:", style, points)
11
    plot.SavePlot("output.png")
12
}
Copied!
在前面的go代码中，你了解了如何使用Glot库及其Glot.SavePlot()函数创建PNG文件。
可以猜到，在编译和执行CSVplot.go之前，需要下载Glot Go库，它需要从Unix shell执行以下命令：
1
$ go get github.com/Arafatk/glot
Copied!
你可以通过查看~/go目录查看当前命令的执行结果:
1
$ ls -l ~/go/pkg/darwin_amd64/github.com/Arafatk/
2
total 240
3
-rw-r--r-- 1 mtsouk staff 119750 Jan 22 22:12 glot.a
4
$ ls -l ~/go/src/github.com/Arafatk/glot/
5
total 120
6
-rw-r--r--  1 mtsouk  staff   1818   Jan  22 22:12 README.md
7
-rw-r--r--  1 mtsouk  staff   6092   Jan  22 22:12 common.go
8
-rw-r--r--  1 mtsouk  staff    552   Jan  22 22:12 common_test.go
9
-rw-r--r--  1 mtsouk  staff   3162   Jan  22 22:12 core.go
10
-rw-r--r--  1 mtsouk  staff    138   Jan  22 22:12 core_test.go
11
-rw-r--r--  1 mtsouk  staff   3049   Jan  22 22:12 function.go
12
-rw-r--r--  1 mtsouk  staff    511   Jan  22 22:12 function_test.go
13
-rw-r--r--  1 mtsouk  staff   4955   Jan  22 22:12 glot.go
14
-rw-r--r--  1 mtsouk  staff    220   Jan  22 22:12 glot_test.go
15
-rw-r--r--  1 mtsouk  staff  10536   Jan  22 22:12 pointgroup.go
16
-rw-r--r--  1 mtsouk  staff    378   Jan  22 22:12 pointgroup_test.go
Copied!
包含要绘制的点的CSV数据文件具有以下格式：
1
$ cat /tmp/dataFile
2
1,2
3
2,3
4
3,3
5
4,4
6
5,8
7
6,5
8
-1,12
9
-2,10
10
-3,10
11
-4,10
Copied!
执行CSVplot.go会产生如下的输出：
1
$ go run CSVplot.go /tmp/doesNoExits
2
Cannot stat /tmp/doesNoExits
3
$ go run CSVplot.gp /tmp/dataFile
4
[[1 2 3 4 5 6 -1 -2 -3 -4] [2 3 3 4 8 5 12 10 10 10]]
Copied!
你可以用下图查看CSVplot.go的结果。 
​

08.7 为什么我们使用二进制格式

08.9 写入文件

Last modified 2yr ago

Copy link