微积分学习笔记（2）：用Go语言画函数图像

本文链接：https://blog.csdn.net/m0_37554486/article/details/133266882

本文介绍了如何使用Go的gonum.org/v1/plot库绘制函数图像，包括创建绘图、生成点、连线并保存为图片的过程，以sin函数为例进行详细说明。

摘要由CSDN通过智能技术生成

使用的Go库

gonum.org/v1/plot
image/color

待绘图函数

直接使用三角函 $s in$ :

func f(x float64) float64 {
	return math.Sin(x)
}

绘图过程

1 创建一个绘图

	p := plot.New()
	assert.NotNil(t, p)
	p.Title.Text = "Function Image"
	p.X.Label.Text = "X"
	p.Y.Label.Text = "Y"

2 生成一些函数的点

	points := make(plotter.XYZs, 100, 100)
	for i := range points {
		x := float64(i) / 10.0
		y := f(x)
		points[i].X = x
		points[i].Y = y
	}

3 将这些点连成线条


	line, err := plotter.NewLine(points)
	assert.Nil(t, err)
	line.Color = color.RGBA{R: 255}
	p.Add(line)

4 将绘图保存到图片


	err = p.Save(4*vg.Inch, 4*vg.Inch, "func_image.png")
	assert.Nil(t, err)

5 绘图效果

在这里插入图片描述

完整源码

函数定义

package funcimage

import "math"

func f(x float64) float64 {
	return math.Sin(x)
}

画图像

package funcimage

import (
	"github.com/stretchr/testify/assert"
	"gonum.org/v1/plot"
	"gonum.org/v1/plot/plotter"
	"gonum.org/v1/plot/vg"
	"image/color"
	"testing"
)

func Test_func_image(t *testing.T) {
	p := plot.New()
	assert.NotNil(t, p)
	p.Title.Text = "Function Image"
	p.X.Label.Text = "X"
	p.Y.Label.Text = "Y"

	points := make(plotter.XYZs, 100, 100)
	for i := range points {
		x := float64(i) / 10.0
		y := f(x)
		points[i].X = x
		points[i].Y = y
	}

	line, err := plotter.NewLine(points)
	assert.Nil(t, err)
	line.Color = color.RGBA{R: 255}
	p.Add(line)

	err = p.Save(4*vg.Inch, 4*vg.Inch, "func_image.png")
	assert.Nil(t, err)
}