Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
152 lines (122 sloc) 3.7 KB

读取CSV文件

CSV文件是纯文本文件。在本节中,你将学习如何读取包含平面点的文本文件,这意味着每一行将包含一对坐标。此外,你还将使用一个名为Glot的外部Go库,它将帮助你创建从CSV文件中读取的点的图表。注意,Glot使用Gnuplot,这意味着你需要在Unix机器上安装Gnuplot才能使用Glot

本章的程序是CSVplot.go,分为五部分。第一部分代码如下:

package main

import (
	"encoding/csv"
	"fmt"
	"github.com/Arafatk/glot"
	"os"
	"strconv"
)

CSVplot.go第二部分代码如下:

func main() {
	if len(os.Args) != 2 {
		fmt.Println("Need a data file!")
		return
	}

	file := os.Args[1]
	_, err := os.Stat(file)
	if err != nil {
		fmt.Println("Cannot stat", file)
		return
	}

在本部分中,你将看到一种使用强大的os.Stat()函数检查文件是否已经存在的技术。

CSVplot.go第三部分代码如下:

	f, err := os.Open(file)
	if err != nil {
		fmt.Println("Cannot open", file)
		fmt.Println(err)
		return
	}
	defer f.Close()

	reader := csv.NewReader(f)
	reader.FieldsPerRecord = -1
	allRecords, err := reader.ReadAll()
	if err != nil {
		fmt.Println(err)
		return
	}

CSVplot.go第四部分代码如下:

	xP := []float64{}
	yP := []float64{}

	for _, rec := range allRecords {
		x, _ := strconv.ParseFloat(rec[0], 64)
		y, _ := strconv.ParseFloat(rec[1], 64)
		xP = append(xP, x)
		yP = append(yP, y)
	}

	points := [][]float64{}
	points = append(points, xP)
	points = append(points, yP)
	fmt.Println(points)

此处,你将字符串转为数字,并追加到二维切片points中。

CSVplot.go最后一部分代码如下:

	dimensions := 2
	persist := true
	debug := false
	plot, _ := glot.NewPlot(dimensions, persist, debug)

	plot.SetTitle("Using Glot with CSV data")
	plot.SetXLabel("X-Axis")
	plot.SetYLabel("Y-Axis")
	style := "circle"
	plot.AddPointGroup("Circle:", style, points)
	plot.SavePlot("output.png")
}

在前面的go代码中,你了解了如何使用Glot库及其Glot.SavePlot()函数创建PNG文件。

可以猜到,在编译和执行CSVplot.go之前,需要下载Glot Go库,它需要从Unix shell执行以下命令:

$ go get github.com/Arafatk/glot

你可以通过查看~/go目录查看当前命令的执行结果:

$ ls -l ~/go/pkg/darwin_amd64/github.com/Arafatk/
total 240
-rw-r--r-- 1 mtsouk staff 119750 Jan 22 22:12 glot.a
$ ls -l ~/go/src/github.com/Arafatk/glot/
total 120
-rw-r--r--  1 mtsouk  staff   1818   Jan  22 22:12 README.md
-rw-r--r--  1 mtsouk  staff   6092   Jan  22 22:12 common.go
-rw-r--r--  1 mtsouk  staff    552   Jan  22 22:12 common_test.go
-rw-r--r--  1 mtsouk  staff   3162   Jan  22 22:12 core.go
-rw-r--r--  1 mtsouk  staff    138   Jan  22 22:12 core_test.go
-rw-r--r--  1 mtsouk  staff   3049   Jan  22 22:12 function.go
-rw-r--r--  1 mtsouk  staff    511   Jan  22 22:12 function_test.go
-rw-r--r--  1 mtsouk  staff   4955   Jan  22 22:12 glot.go
-rw-r--r--  1 mtsouk  staff    220   Jan  22 22:12 glot_test.go
-rw-r--r--  1 mtsouk  staff  10536   Jan  22 22:12 pointgroup.go
-rw-r--r--  1 mtsouk  staff    378   Jan  22 22:12 pointgroup_test.go

包含要绘制的点的CSV数据文件具有以下格式:

$ cat /tmp/dataFile
1,2
2,3
3,3
4,4
5,8
6,5
-1,12
-2,10
-3,10
-4,10

执行CSVplot.go会产生如下的输出:

$ go run CSVplot.go /tmp/doesNoExits
Cannot stat /tmp/doesNoExits
$ go run CSVplot.gp /tmp/dataFile
[[1 2 3 4 5 6 -1 -2 -3 -4] [2 3 3 4 8 5 12 10 10 10]]

你可以用下图查看CSVplot.go的结果。 ""

You can’t perform that action at this time.