matplotlib之添加子图

import matplotlib.pyplot as plt
import numpy as np

rad = np.arange(0,np.pi*2 + 0.1 , 0.1)

#sin
y = np.sin(rad)

#画布
p1 = plt.figure(figsize = (8,6),dpi = 80) #figsize画布大小 ， dpi指分辨率
plt.title('sin/cos')

#创建第一个子图sin函数
p1.add_subplot(2,1,1)
plt.plot(rad,y)
plt.legend(['sin'])

xticks1 = np.array([0,np.pi/2,np.pi,3*np.pi/2,np.pi*2])
#plt.xticks(xticks1)

yticks1 = np.sin(xticks1)
#print(yticks1)
plt.yticks(yticks1])

#添加第二个子图cos函数
p1.add_subplot(2,1,2)  #二行一列第二个图
plt.plot(rad, np.cos(rad))
plt.legend(['cos'])
plt.yticks([0,-1,1])

plt.savefig('sin_cos.png')
plt.show()