要让程序自动将图表保存到文件中,可将对plt.show()的调用替换为plt.savefig()的调用:
plt.savefig('squares_plot.png',bbox_inches='tight')
第一个实参指定要以什么样的文件名保存图表,这个文件将存储到执行程序所在的目录中;
第二个实参指定将图表多余的空白区域裁减掉。如果要保留图表周围多余的空白区域,可省略这个实参。
#!/usr/bin/env python
# -*- coding:utf-8 -*-
import matplotlib.pyplot as plt
x_values = list(range(1, 501))
y_values = [x**2 for x in x_values]
plt.scatter(x_values, y_values, s=50, c=y_values, cmap=plt.cm.Blues, edgecolors='none')
# 设置图表标题, 并给坐标轴加上标签
plt.title("Square Numbers", fontsize=24)
plt.xlabel("Value", fontsize=14)
plt.ylabel("Square of Value", fontsize=14)
# 设置刻度标记的大小
plt.tick_params(axis='both', which='major', labelsize=14)
plt.savefig('squares_plot.png', bbox_inches='tight')