Golangでグラフを描く

はじめに

golangでグラフを描く時，つまづいたので備忘録として残しておきます．
どうやらgonum plotが便利らしい．
しかし，古い記事だとgo getの場所がgoogle codeを参照しているようだ．
現在は，githubに移行しているので以下を実行．

go get gonum.org/v1/plot/...

二つのデータ群をplot

(8.0, 2.0)と(3.0, 6.0)を軸にそれぞれclass1,class2として100個,ガウス分布に従って，学習データを生成したものをplotする．
以下がそのコードである．図の生成からplotまで，コメント文を参照してください．

package main

import (
    "image/color"![plot.png](https://qiita-image-store.s3.amazonaws.com/0/123283/3a191005-7612-21c4-88ef-aa6b3f3c5e29.png)

    "math/rand"
    "time"

    "gonum.org/v1/plot"
    "gonum.org/v1/plot/plotter"
    "gonum.org/v1/plot/vg"
)

func main() {

    // 図の生成
    p, err := plot.New()
    if err != nil {
        panic(err)
    }
    //label
    p.Title.Text = "Points Example"
    p.X.Label.Text = "X"
    p.Y.Label.Text = "Y"

    // 補助線
    p.Add(plotter.NewGrid())

    //クラス1
    x1, y1 := 8.0, 2.0

    //クラス2
    x2, y2 := 3.0, 6.0

    //各クラスのサンプル
    n := 200

    // 散布図の作成
    plot1, err := plotter.NewScatter(randomPoints(n, x1, y1))
    if err != nil {
        panic(err)
    }

    plot2, err := plotter.NewScatter(randomPoints(n, x2, y2))
    if err != nil {
        panic(err)
    }

　   //色を指定する．
    plot1.GlyphStyle.Color = color.RGBA{R: 255, B: 128, A: 55}
    plot2.GlyphStyle.Color = color.RGBA{R: 155, B: 128, A: 255}

    //plot1,plot2をplot
    p.Add(plot1)
    p.Add(plot2)

    //label
    p.Legend.Add("class1", plot1)
    p.Legend.Add("class2", plot2)

    // 座標範囲
    p.X.Min = 0
    p.X.Max = 10
    p.Y.Min = 0
    p.Y.Max = 10

    // plot.pngに保存
    if err := p.Save(6*vg.Inch, 6*vg.Inch, "plot.png"); err != nil {
        panic(err)
    }

}

//ガウス分布
func random(axis float64) float64 {
    //分散
    dispersion := 1.0
    rand.Seed(time.Now().UnixNano())
    return rand.NormFloat64()*dispersion + axis
}

//学習データの生成
func randomPoints(n int, x, y float64) plotter.XYs {
    pts := make(plotter.XYs, n)
    for i := range pts {
        pts[i].X = random(x)
        pts[i].Y = random(y)
    }
    return pts
}

結果

実際にplotすることができた．

参考文献

Go でグラフを plot するパッケージを試した
gonum