はじめに
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"
"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することができた．