首页 > 代码库 > 利用scatter()绘制颜色映射的二次方曲线

利用scatter()绘制颜色映射的二次方曲线

程序如下:

 1 import matplotlib.pyplot as plt
 2 
 3 x_value = http://www.mamicode.com/list(range(1, 1001))
 4 y_value = http://www.mamicode.com/[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值的范围,避免重复劳动。 

 

利用scatter()绘制颜色映射的二次方曲线