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的结果。
Last modified 2yr ago
Copy link