pandas画图-【老鱼学pandas】
本节主要讲述如何把pandas中的数据用图表的方式显示在屏幕上,有点类似在excel中显示图表。
安装matplotlib
为了能够显示图表,首先需要安装matplotlib库,安装方法如下:
pip3 install matplotlib
显示图表
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# 产生1000个随机数
data = pd.Series(np.random.rand(1000))
# 画图表
data.plot()
# 显示图表
plt.show()
在上面的代码中,我们首先创建1000个随机数序列点,然后对这1000个点进行画图表,最后调用plt.show()显示图表。
显示的结果为:
在matplotlib中,如果想要显示图表,需要使用plt.plot()函数,但在pandas中,我们只要对序列数据调用plot()函数就能显示其所拥有的数据,是不是很简单。
显示矩阵数据
显示矩阵中的数据,就会把每列当做一个数据序列进行显示,例如:
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# 产生1000个4列正态分布的随机数
data = pd.DataFrame(np.random.randn(1000, 4))
# 对其中的数据进行累加,目的是为了能够使例子中显示的曲线有上升波动效果
data = data.cumsum()
# 打印出前几行的数据,方便我们调试其中的数据值
print(data.head())
# 画图表
data.plot()
# 显示图表
plt.show()
其中显示的头几行数据值为:
0 1 2 3
0 0.596323 0.195563 0.037300 0.900385
1 0.186813 2.217155 0.516096 0.623348
2 1.147338 2.362926 2.597531 0.960079
3 1.571033 2.910459 0.947950 0.609487
4 1.783988 3.702685 -0.358119 2.341726
显示的图表为:
还是挺漂亮的图表,有点像股市走势图。:)
只是其中的0,1,2,3这几个列名不大好看,我们可以修改产生数据集中的列名,这样,在显示图表时就能按照数据集中的列名进行显示了。
比如,我们想修改数据集中的列名显示为A、B、C、D:
data = pd.DataFrame(np.random.randn(1000, 4), columns=list("ABCD"))
这样显示为:
显示散点图
可以用scatter()函数来显示散点图,但其使用方式跟显示图表的有点不一样,主要是此函数是在plot属性中的方法,需要注意一下:
data.plot.scatter(x='A', y='B')
完整的代码为:
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# 产生1000个4列正态分布的随机数
data = pd.DataFrame(np.random.randn(1000, 4), columns=list("ABCD"))
# 打印出前几行的数据,方便我们调试其中的数据值
print(data.head())
# 显示散点图
data.plot.scatter(x='A', y='B')
# 显示图
plt.show()
显示图为:
在scatter函数中还可以添加额外的属性,比如可以通过color来指定颜色,通过label来指定标签,我们来看一下:
data.plot.scatter(x='A', y='B', color='DarkBlue', label="Class 1")
显示的图为:
如果我们想要在一张图中再画一下另一个散点图,比如,在上面这个例子中,我们已经显示了横坐标是A纵坐标是B的散点图,现在我们还想把横坐标是A,纵坐标是C的数据也显示在其中,该怎么进行呢?
# 显示散点图
pic1 = data.plot.scatter(x='A', y='B', color='DarkBlue', label="Class 1")
data.plot.scatter(x='A', y='C', color='DarkGreen', label='Class2', ax=pic1)
# 显示图
plt.show()
图片显示为:
更多图形
更多的图形还有:条形图, 直方图, 盒须图, 饼图等,相应的函数名有:bar, hist, box, kde, area, hexbin, pie等(前面不是一一对应的,有兴趣的读者可以自己去试试看。)
出处:http://www.cnblogs.com/dreampursuer/
本文版权归作者所有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。