plot绘制子图

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/ThinkPet/article/details/83013919

# -*- coding: utf-8 -*-
"""
Created on Thu Oct 11 15:38:42 2018

@author: Administrator
"""

import numpy as np
rad = np.arange(0,np.pi*2,0.01)
print('范围0-2pi，步长0.01的数组\n',rad)
print(type(rad))
import matplotlib.pyplot as plt
#设置画布，创建子图
p1 = plt.figure(figsize=(8,6) ,dpi=80)  #画布 长宽8:6,分辨率=80
#第一个子图ax1
ax1 = p1.add_subplot(2,1,1)  #创建2行1列的子图，开始绘制第一个子图ax1
plt.title('lines')
plt.xlabel('x')  #设置x轴名称
plt.ylabel('y')  #设置y轴名称
plt.xlim((0,1))  #设置x轴范围
plt.ylim((0,1))  #设置y轴范围
plt.xticks([0,0.2,0.4,0.6,0.8,1]) #设置x轴刻度
plt.yticks([0,0.2,0.4,0.6,0.8,1]) #设置y轴刻度
plt.plot(rad,rad**2)  #绘制y=x^2曲线
plt.plot(rad,rad**4)  #绘制y=x^4曲线
plt.legend( ['y=x^2' , 'y=x^4'])  #添加注解

#第2个子图ax2
ax2 = p1.add_subplot(2,1,2)  #创建2行1列的子图，开始绘制第2个子图ax2
plt.title('sin/cos')
plt.xlabel('rad')  #设置x轴名称
plt.ylabel('value')  #设置y轴名称
plt.xlim((0,np.pi*2))  #设置x轴范围
plt.ylim((-1,1))  #设置y轴范围
plt.xticks([0,   np.pi/2,   np.pi  ,  np.pi*1.5   ,np.pi*2]) #设置x轴刻度
plt.yticks([-1,   -0.5,      0,         0.5,         1]) #设置y轴刻度
plt.plot(rad,   np.sin(rad))  #绘制y=x^2曲线
plt.plot(rad,   np.cos(rad))  #绘制y=x^4曲线
plt.legend( ['sin' , 'cos'])  #添加注解

#保存并展示图片
plt.savefig('./sincos.png')
plt.show()