Python 可否 保存figure

import matplotlib.pyplot as plt

#这里应该把trace plot 出来，同时，还应该把峰值给plot出来；
plt.figure()
plt.plot(sigAv, color='red', label='dis')
plt.plot(peaks, sigAv[peaks], "x")
plt.savefig(output)
plt.show()