import numpy as np
import matplotlib.pyplot as plt
x1 = np.linspace (0.0, 5.0) # 画图一
x2 = np.linspace (0.0, 2.0) # 画图二
y1 = np.cos (2 * np.pi * x1) * np.exp (-x1)
y2 = np.cos (2 * np.pi * x2)
plt.subplot (2, 1, 1) # 面板设置成2行1列,并取第一个(顺时针编号)
plt.plot (x1, y1, ‘yo-’) # 画图,染色
plt.title (‘A tale of 2 subplots’)
plt.ylabel (‘Damped oscillation’)
plt.subplot (2, 1, 2) # 面板设置成2行1列,并取第二个(顺时针编号)
plt.plot (x2, y2, ‘r.-’) # 画图,染色
plt.xlabel (‘time (s)’)
plt.ylabel (‘Undamped’)