欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

利用matplotlib画函数图像

程序员文章站 2022-03-31 14:53:08
...

主要步骤:

准备数据
配置画板
绘制
保存
代码如下:

import numpy as np
import matplotlib.pyplot as pt
from matplotlib import font_manager

# pt.rcParams["font.sans-serif"] = ['SimHei']
# 设置中文字体显示
my_font = font_manager.FontProperties(fname="C:\Windows\Fonts\simsun.ttc")
# 准备数据
X = np.linspace(-np.pi, np.pi * 2, 1000)
# 配置画板,figsize用来设置图例的大小
pt.figure(figsize=(6,3),facecolor="yellow")
# subplot的第一个参数表示行,第二个参数表示列,第三个参数表示图例中的**位置
pt.subplot(1,1,1)
# 准备Y数据
Y_x2 = X1**2
Y_sin = np.sin(X)

# 标题,X坐标和Y坐标名称
pt.xlabel(u"X数值",fontproperties=my_font)
pt.ylabel(u"Y数值",fontproperties=my_font)
pt.title(u"函数图像",fontproperties=my_font,fontsize=20)

# 设置X,Y轴上下限
pt.xlim(-np.pi,np.pi)
pt.ylim(-1,1)

# 设置关键刻度
pt.xticks([-np.pi,-np.pi/2.0,0,np.pi/2,np.pi])
# 添加文字,第一个参数为X坐标,第二个参数是Y轴坐标,以数据的刻度为基准
pt.text(0.0,-0.5,'don`s move',fontdict={'size':'16','color':'b'})
# 添加标注,xy是标注箭头要指的点,xytext:描述信息的坐标,arrowprops箭头的样式
pt.annotate('note!!',xy=(-np.pi/2.0,-1),xytext=(-np.pi/2,-0.25),fontsize=16,arrowprops=dict(facecolor='black',shrink=0.01))

# 保存到文件
pt.savefig("test2.jpg")

# 图例
pt.plot(X1, Y_x2,c="g",lw=3,label=u"x2函数")
# 直线-,虚线--,点线:,点划线-.
pt.plot(X, Y_sin,c="r",lw=5,ls="-.",label=u"sin函数")
# 添加图例
pt.legend(prop=my_font)

pt.show()