#绘制简单的折线图
#导入模块pyplot,指定别名plt,以免反复输入pyplot
import matplotlib.pyplot as plt
#plt.subplot(211) #要生成两行一列,这是第一个图plt.subplot('行','列','编号')
input_value=[1,2,3,4,5]
squares = [1, 4, 9, 16, 25]
plt.plot(input_value,squares,linewidth=5) #数linewidth()决定了plot()绘制的线条的粗细
# 设置图表标题,并给坐标轴加上标签
plt.title("Square Numbers", fontsize=24) #参数fontsize指定了图表中文字的大小
plt.xlabel("Value", fontsize=14)
plt.ylabel("Square of Value", fontsize=14)
# 设置刻度标记的大小
plt.tick_params(axis='both', labelsize=14) #函数tick_params()设置刻度的样式,指定的实参将影响x轴和y轴上的刻度(axes='both'),刻度标记的字号设置为14(labelsize=14)
#保存,第一个实参指定要以什么样的文件名保存图表,第二个实参指定将图表多余的空白区域裁剪掉
plt.savefig('squares_plot.png', bbox_inches='tight')
#plt.show()
#plt.subplot(2,1,2)
#使用 scatter()绘制散点图并设置其样式
x_values = list(range(1, 5001))
y_values = [x**3 for x in x_values]
#删除数据点的轮廓,调用scatter()时传递实参edgecolor='none'
#向scatter()传递参数c,并将其设置为要使用的颜色的名称
#还可以使用RGB颜色模式自定义颜色, c=(0, 0, 0.8),值越接近0,指定的颜色越深,值越接近1,指定的颜色越浅。
#plt.scatter(x_values, y_values, c=(0, 0, 0.8), edgecolor='none', s=40)
#plt.scatter(x_values, y_values, c='red', edgecolor='none', s=40)
#颜色映射,参数c设置成了一个y值列表,并使用参数cmap告诉pyplot使用哪个颜色映射。将y值较小的点显示为浅蓝色,并将y值较大的点显示为深蓝色
plt.scatter(x_values, y_values, c=y_values, cmap=plt.cm.Blues, edgecolor='none', s=40)
# 设置图表标题并给坐标轴加上标签
plt.title("Square Numbers", fontsize=24)
plt.xlabel("Value", fontsize=14)
plt.ylabel("Square of Value", fontsize=14)
# 设置每个坐标轴的取值范围
#plt.axis([0, 1100, 0, 1100000])
# 设置刻度标记的大小
plt.tick_params(axis='both', which='major', labelsize=14)
plt.savefig('scatter_plot.jpg', bbox_inches='tight')
plt.show()