利用matplotlib画函数图像

主要步骤:

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

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()

你可能感兴趣的:(python数据分析)