读取CSV文件

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

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

  1. package main
  3. import (
  4.     "encoding/csv"
  5.     "fmt"
  6.     "github.com/Arafatk/glot"
  7.     "os"
  8.     "strconv"
  9. )

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

  1. func main() {
  2.     if len(os.Args) != 2 {
  3.         fmt.Println("Need a data file!")
  4.         return
  5.     }
  7.     file := os.Args[1]
  8.     _, err := os.Stat(file)
  9.     if err != nil {
  10.         fmt.Println("Cannot stat", file)
  11.         return
  12.     }

在本部分中，你将看到一种使用强大的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()
  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.     }

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

  1.     xP := []float64{}
  2.     yP := []float64{}
  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.     }
  11.     points := [][]float64{}
  12.     points = append(points, xP)
  13.     points = append(points, yP)
  14.     fmt.Println(points)

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

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

  1.     dimensions := 2
  2.     persist := true
  3.     debug := false
  4.     plot, _ := glot.NewPlot(dimensions, persist, debug)
  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. }

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

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

  1. $ go get github.com/Arafatk/glot

你可以通过查看~/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

包含要绘制的点的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

执行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]]

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