有深度学习需求的小伙伴请点击原文链接【教程】第三章:图例各类图形,在线调试代码,玩转数据分析!
本章节是教程中最为重要的章节,本节我们将开始讲述各种不同的plot的方式。之前我们讲到了如何plot线,今天我们讲述如何画出散点图、柱状图、等高线图和image图片。学完本章之后,小伙伴们就基本能够应对工作和学习中大部分的画图场景啦。
本章知识点归纳如下:
1.散点图:plt.scatter()
2.柱状图:plt.bar()
3.等高线图:plt.contourf()
4.在等高线图中增加label:plt.clabel()
5.矩阵画图:plt.imshow()
6.在随机矩阵图中增加colorbar:plt.colorbar()
散点图
首先,先引入matplotlib.pyplot简写作plt,再引入模块numpy用来产生一些随机数据。
import matplotlib.pyplot as plt
import numpy as np
n = 1024 # data size
X = np.random.normal(0, 1, n) # 每一个点的X值
Y = np.random.normal(0, 1, n) # 每一个点的Y值
T = np.arctan2(Y,X) # for color value
1.数据生成
生成1024个呈标准正态分布的二维数据组 (平均数是0,方差为1) 作为一个数据集,并图像化这个数据集。每一个点的颜色值用T来表示:
2.画图:
数据集生成完毕,现在来用 plt.scatter 画出这个点集,输入X和Y作为location,size=75,颜色为T,color map用默认值,透明度alpha 为 50%。 x轴显示范围定位(-1.5,1.5),并向xtick()函数传入空集()来隐藏x坐标轴,y轴同理:
plt.scatter(X, Y, s=75, c=T, alpha=.5)
plt.xlim(-1.5, 1.5)
plt.xticks(()) # ignore xticks
plt.ylim(-1.5, 1.5)
plt.yticks(()) # ignore yticks
plt.show()