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

Posted 2020-09-18

程序如下：

 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值的范围，避免重复劳动。