matplotlib绘制2D图表

ipython notebook 代码

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
%matplotlib inline


#设置全局横纵字体大小
mpl.rcParams['xtick.labelsize'] = 24
mpl.rcParams['ytick.labelsize'] = 24

np.random.seed(42)

#x轴的采样点
x = np.linspace(0, 5, 100)

#通过下面曲线加上噪声生成数据，所以拟合模型就用y了
y = 2*np.sin(x) + 0.3*x**2
y_data = y + np.random.normal(scale=0.3, size=100)

#figure()指定图表名称
plt.figure('data')

#'.'标明画散点图，每个散点的形状是个圆
plt.plot(x, y_data, '.')

#画模型的图，plot函数默认画连线图
plt.figure('model')
plt.plot(x, y)

#两个图画一起
plt.figure('data & model')

#k：颜色 lw：线的宽度 第三个参数可以指定线型，eg：'r--'表示红色虚线
plt.plot(x, y, 'k', lw=3)

#生成散点图
plt.scatter(x, y_data)

#保存
plt.savefig('result.png')

#显示图在屏幕
plt.show()

ipython运行结果：