利用scatter()绘制颜色映射的二次方曲线
程序如下:
1 import matplotlib.pyplot as plt 2 3 x_value = list(range(1, 1001)) 4 y_value = [x**2 for x in x_value] 5 6 plt.scatter(x_value, y_value, c=y_value, cmap=plt.cm.summer, 7 edgecolor='none', s=2) 8 9 plt.title("Square Number", fontsize=14) 10 plt.xlabel("Value", fontsize=8) 11 plt.ylabel("Square of Value", fontsize=8) 12 13 plt.axis([0, 1100, 0, 1100000]) 14 plt.savefig('square_plot.png',bbox_inches='tight')
程序运行后,图将保持在程序所在的文件夹内,最终图的效果如下(因为夏天快来了,所以颜色映射我选的是summer):
小结:
① 颜色映射在数据挖掘中可以突出数据的变化规律,比如上图中黄色表示数值较大,绿色表示数据较小,且这种变化是连续的而非离散。
② 使用edgecolor可以使得曲线较为平滑,利于后期演示。
③ 利用for x in x_value历编整个x值的范围,避免重复劳动。