http://www.cnblogs.com/batteryhp/p/5025772.html
python有许多可视化工具,本书主要讲解matplotlib。matplotlib是用于创建出版质量图表的桌面绘图包(主要是2D方面)。matplotlib的目的是为了构建一个MATLAB式的绘图接口。本书中的大部分图都是用它生成的。除了图形界面显示,还可以把图片保存为pdf、svg、jpg、png、gif等形式。
1、matplotlib API入门
Ipython可以用close()关闭界面。
Figure和Subplot
matplotlib的图像都位于Figure对象中。用plt.figure创建一个新的Figure。
import numpy as np import pandas as pd import matplotlib.pyplot as plt ‘‘‘ #plt.plot(np.arange(10)) fig = plt.figure() #plt.show() #figsize 有一些重要的选项,特别是figsize,规定的是图片保存到磁盘时具有一定大小的纵横比。 #plt.gcf()即可得到当前Figure的引用 ax1 = fig.add_subplot(2,2,1) ax2 = fig.add_subplot(2,2,2) ax3 = fig.add_subplot(2,2,3) plt.plot(np.random.randn(50).cumsum(),‘k--‘) #fig.add_subplot 返回的对象是AxesSubplot对象,下面调用就可以了 _ = ax1.hist(np.random.randn(100),bins = 20,color = ‘k‘,alpha = 0.3) ax2.scatter(np.arange(30),np.arange(30) + 3 * np.random.randn(30)) plt.show() ‘‘‘ #由于Figure 和 subplot是一件非常常见的任务,于是出现了更为方便的方法(plt.subplots ),它可以创建一个新的Figure, #并返回一个含有已创建的subplot对象的Numpy数组 fig,axes = plt.subplots(2,3) #print fig print axes[0][0] #axes[0][0].hist(np.random.randn(100),bins = 20,color = ‘k‘,alpha = 0.3) plt.show() #这是非常实用的,因为可以轻松地对axes数组进行索引,就好像一个是一个二维数组一样,例如 #axes[0,1].还可以通过sharex和sharey指定subplot具有相同的x轴和y轴。在比较相同范围的数据时,这是 #非常实用的,否则matplotlib会自动缩放各图表的界限。
看一下subplots的作用:
pyplot.subplots的选项还有:
上面的**fig_k可以有很多的参数,文档中有更多的内容。
调整subplot周围的间距
默认情况下,matplotlib会在subplot外围留下一定的边距,并在subplot之间留下一定的间距。间距和图像的高度和宽度有关,会自动调整。利用Figure的subplots——adjust方法可以修改间距,因此,它是一个顶级函数。
import numpy as np import pandas as pd import matplotlib.pyplot as plt subplots_adjust(left = None,bottom = None,right = None,top = None,wspace = None,hspace = None) #wspace和space用于控制宽度和高度的百分比,可以用做subplot之间的间距,下面是个例子: ‘‘‘ fig,ax = plt.subplots(2,2,sharex = True,sharey = True) for i in range(2): for j in range(2): ax[i,j].hist(np.random.randn(500),bins = 50,color = ‘k‘,alpha = 0.5) plt.subplots_adjust(wspace = 0.5,hspace = 0.5) plt.show() #matplotlib不会检查标签的重叠(确实是这样)。
# -*- encoding: UTF-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt fig,ax = plt.subplots(2,2) #cecece是白色…… ax[0,0].plot(np.arange(10),linestyle = ‘--‘,color = ‘#CECECE‘) #线上面还可以添加一些标记(marker),以强调实际的数据点。由于matplotlib创建的是连续的线形图,因此有时可能不太容易看到真实点的位置,标记可以放到格式字符串中,但是标记类型和线性必须在颜色的后面 ax[0,1].plot(np.random.randn(30).cumsum(),‘ko--‘) ax[1,0].plot(np.random.randn(30).cumsum(),color = ‘k‘,linestyle = ‘--‘,marker = ‘o‘) #在线型图中,非实际数据点默认是按照线性插值的,可以通过drawstyle选项修改这一点。 data = np.random.randn(30).cumsum() ax[1,1].plot(data,‘ko--‘) ax[1,1].plot(data,‘k--‘,drawstyle = ‘steps-post‘) plt.show()
注意上面的drawstyle选项可以规定点与点之间的连接方式,或者说是插值方式,结果为:
设置标题、轴标签、刻度以及刻度标签
# -*- encoding: UTF-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt import numpy.random as npr fig = plt.figure() ax = fig.add_subplot(1,1,1) ax.plot(npr.randn(1000).cumsum()) #要想修改x的刻度,最简单的方法就是使用set_xticks和set_xticklabels.前者告诉matplotlib将 #刻度放在数据范围中的哪些位置,默认情况下,这些位置就是刻度标签。但是可以使用set_xticklabels将#任何其他的值用作标签 ticks = ax.set_xticks([0,250,500,700,900,1000]) #下面的totation是规定旋转角度 labels = ax.set_xticklabels([‘a‘,‘b‘,‘c‘,‘d‘,‘e‘,‘f‘],rotation = 30,fontsize = ‘small‘) #可以为x轴设置名称 ax.set_xlabel(‘Stages‘) plt.show()
图例
# -*- encoding: UTF-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt import numpy.random as npr from datetime import datetime #添加图例 fig = plt.figure() ax = fig.add_subplot(1,1,1) ax.plot(npr.randn(1000).cumsum(),‘k‘,label = ‘one‘) ax.plot(npr.randn(1000).cumsum(),‘k--‘,label = ‘two‘) ax.plot(npr.randn(1000).cumsum(),‘k.‘,label = ‘three‘) ax.legend(loc = ‘best‘) plt.show()
注解与绘图
# -*- encoding: UTF-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt import numpy.random as npr from datetime import datetime fig = plt.figure() ax = fig.add_subplot(1,1,1) data = pd.read_csv(‘E:\\spx.csv‘,index_col = 0,parse_dates = True) spx = data[‘SPX‘] spx.plot(ax = ax,style = ‘k-‘) crisis_data = [ (datetime(2007,10,11),‘Peak of bull market‘), (datetime(2008,3,12),‘Bear Stearns Fails‘), (datetime(2008,9,15),‘Lehman Bankruptcy‘) ] for date,label in crisis_data: ax.annotate(label,xy = (date,spx.asof(date) + 50), xytext = (date,spx.asof(date) + 200), arrowprops = dict(facecolor = ‘black‘), horizontalalignment = ‘left‘,verticalalignment = ‘top‘) ax.set_xlim([‘1/1/2007‘,‘1/1/2011‘]) ax.set_ylim([600,1800]) ax.set_title(‘Important dates in 2008-2009 finacial crisis‘) plt.show() #更多关于注解的示例,请看文档 #图形的绘制要麻烦些,有一些常见的图形的对象,这些对象成为块(patch) #如Rectangle 和 Circle,完整的块位于matplotlib.patches #要绘制图形,需要创建一个块对象shp,然后通过ax.add_patch(shp)将其添加到subplot中 fig = plt.figure() ax = fig.add_subplot(1,1,1) rect = plt.Rectangle((0.2,0.75),0.4,0.15,color = ‘k‘,alpha = 0.3) circ = plt.Circle((0.7,0.2),0.15,color = ‘b‘,alpha = 0.3) pgon = plt.Polygon([[0.15,0.15],[0.35,0.4],[0.2,0.6]],color = ‘g‘,alpha = 0.5) ax.add_patch(rect) ax.add_patch(circ) ax.add_patch(pgon) plt.show()
将图表保存到文件
# -*- encoding: UTF-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt import numpy.random as npr from datetime import datetime from io import StringIO #将图标保存到文件 #savefig函数可以保存图形文件,不同的扩展名保存为不同的格式 fig = plt.figure() ax = fig.add_subplot(1,1,1) rect = plt.Rectangle((0.2,0.75),0.4,0.15,color = ‘k‘,alpha = 0.3) circ = plt.Circle((0.7,0.2),0.15,color = ‘b‘,alpha = 0.3) pgon = plt.Polygon([[0.15,0.15],[0.35,0.4],[0.2,0.6]],color = ‘g‘,alpha = 0.5) ax.add_patch(rect) ax.add_patch(circ) ax.add_patch(pgon) #注意下面的dpi(每英寸点数)和bbox_inches(可以剪除当前图标周围的空白部分)(确实有效) #plt.savefig(‘pic.jpg‘,dpi = 100,bbox_inches = ‘tight‘) #不一定save到文件中,也可以写入任何文件型对象,比如StringIO: buffer = StringIO() plt.savefig(buffer) plot_data = buffer.getvalue() #这对Web上提供动态生成的图片是很实用的 #plt.show()
savefig的一些选项:
matplotlib配置
matplotlib的一些属性是可以设置的,比如图像大小、subplot边距、配色方案、字体大小、网格类型等。有两种方式进行操作。第一种是Python变成方式,即利用rc方法。比如:
plt.rc(‘figure‘,figsize = (10,10))
rc的第一个参数是希望自定义的对象,比如‘figure’、‘axes’、‘xtick’、‘ytick’、‘grid’、‘legend’等。其后可以跟上一系列的关键字参数。最简单的就是写成一个字典:
font_options = {‘family‘:‘monospace‘, ‘weight‘:‘bold‘, ‘size‘:‘small‘} plt.rc(‘font‘,**font_options)
matplotlibrc是配置文件,定义好以后每次加载就会用设置的参数。
2、pandas中的绘图函数
matplotlib是一种比较低级的工具,需要将各种组件组合好:数据展示(线型图、柱状图等)、图例、标题、刻度标签以及注解。这是因为制作一张图表一般需要用到多个对象。在pandas中,会省事不少。pandas能够利用DataFrame的对象特点创建标准图表的高级绘图方法。作者说pandas在线文档时最好的学习工具,书上的代码可能过时了。
线型图
#-*- encoding:utf-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt from pandas import Series,DataFrame s = Series(np.random.randn(10).cumsum(),index = np.arange(0,100,10)) #该Series对象的索引会被传给matplotlib,并绘制X轴。 #可以用use_index = False 禁用该功能 s.plot(use_index = False) #X轴的刻度和界限可以通过xticks和xlim选项进行调节,Y轴通过xticks和ylim调节 plt.show() #pandas的大部分方法都有一个可选的ax参数,可以是一个subplot对象。这可以 #使在网格中更为灵活地处理subplot的位置。 #DataFrame的plot方法会在一个subplot中为各列绘制线型图,并自动添加图例 df = DataFrame(np.random.randn(10,4).cumsum(0), columns = [‘A‘,‘B‘,‘C‘,‘D‘], index = np.arange(0,100,10)) df.plot() plt.show()
下面把参数贴一下:
DataFrame还有一些对列进行处理的参数:
自下面开始就有一些专门的图形,绘制的时候可以与R语言进行对比:http://www.cnblogs.com/batteryhp/p/4733474.html。
柱状图
#-*- encoding:utf-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt from pandas import Series,DataFrame #生成的线形图中代码加上kind = ‘bar’(垂直柱图) 或者 (水平)kind = ‘barh’(水平柱图) #Series和DataFrame的索引被用作X(bar)或者Y(barh)的刻度 fig,axes = plt.subplots(2,1) data = Series(np.random.randn(16),index = list(‘abcdefghijklmnop‘)) data.plot(kind = ‘barh‘,ax = axes[0],color = ‘k‘,alpha = 0.7) data.plot(kind = ‘bar‘,ax = axes[1],color = ‘k‘,alpha = 0.7) #DataFrame会按照行对数据进行分组 df = DataFrame(np.random.randn(6,4),index = [‘one‘,‘two‘,‘three‘,‘four‘,‘five‘,‘six‘], columns = pd.Index([‘A‘,‘B‘,‘C‘,‘D‘],name = ‘Genus‘)) #注意这里的name会被用作图例的标题,因为,这本来就是列的名字 print df df.plot(kind = ‘bar‘) plt.show() #这里的stacked是标明画累计柱图 df.plot(kind = ‘bar‘,stacked = True,alpha = 0.5) plt.show() #Series的value_counts可以用来显示Series中各值的频数(实验证明) s = Series([1,2,2,3,4,4,4,5,5,5]) s.value_counts().plot(kind = ‘bar‘) plt.show()
下面看一个例子:
#-*- encoding:utf-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt from pandas import Series,DataFrame #下面是一个例子:做一张堆积柱状图来显示每天各种聚会规模的数据点百分比 tips = pd.read_csv(‘E:\\tips.csv‘) party_counts = pd.crosstab(tips.day,tips.size) print party_counts party_counts = party_counts.ix[:,2:5] #然后进行归一化是各行和为1 party_pcts = party_counts.div(party_counts.sum(1).astype(float),axis = 0) print party_pcts party_pcts.plot(kind = ‘bar‘,stacked = True) plt.show()
直方图和密度图
#-*- encoding:utf-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt from pandas import Series,DataFrame #绘制小费百分比直方图 tips = pd.read_csv(‘E:\\tips.csv‘) tips[‘tip_pct‘] = tips[‘tip‘] / tips[‘total_bill‘] #bins规定一共分多少个组 tips[‘tip_pct‘].hist(bins = 50) plt.show() #与此相关的是密度图:他是通过计算“可能会产生观测数据的连续概率分布的估计” #而产生的。一般的过程将该分布金思维一组核(诸如正态之类的较为简单的分布)。 #此时的密度图称为KDE图。kind = ‘kde’即可。 tips[‘tip_pct‘].plot(kind = ‘kde‘) plt.show() #显然,直方图和密度图经常会在一起出现 comp1 = np.random.normal(0,1,size = 200) comp2 = np.random.normal(10,2,size = 200) values = Series(np.concatenate([comp1,comp2])) print values values.hist(bins = 100,alpha = 0.3,color = ‘k‘,normed = True) values.plot(kind = ‘kde‘,style = ‘k--‘) plt.show()
散布图
散布图(scantter plot)是观察两个一维数据序列之间的关系的有效手段。matplotlib中的scantter方法是绘制散布图的主要方法。
#-*- encoding:utf-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt from pandas import Series,DataFrame #下面加载macrodata中的数据集,选择其中几列并计算对数差 macro = pd.read_csv(‘E:\\macrodata.csv‘) data = macro[[‘cpi‘,‘m1‘,‘tbilrate‘,‘unemp‘]] #这里的diff函数是用来计算相邻两数只差,对每一列,后一个数减前一个数 trans_data = np.log(data).diff().dropna() #print np.log(data).head() #print np.log(data).diff().head() print trans_data.head() plt.scatter(trans_data[‘m1‘],trans_data[‘unemp‘]) plt.title(‘Changes in log %s vs. log %s‘%(‘m1‘,‘unemp‘)) plt.show() #画散布图矩阵式很有意义的pandas提供了scantter_matrix函数来创建散步矩阵 #关于 diagonal 参数,是为了不让对角线上的图形(自己和自己的散布图)显示为一条直线而设置的关于这种数据的某些图形显示 #比如 diagonal = ‘kde‘就是画密度图且核为kde,若diagonal=‘hist‘,则为直方图 pd.scatter_matrix(trans_data,diagonal = ‘kde‘,color = ‘k‘,alpha = 0.3) pd.scatter_matrix(trans_data,diagonal = ‘hist‘,color = ‘k‘,alpha = 0.3) plt.show()
绘制地图:图形化显示海地地震危机数据
这是一个例子。
#-*- encoding:utf-8 -*- import numpy as np import pandas as pd import matplotlib.pyplot as plt from pandas import Series,DataFrame from mpl_toolkits.basemap import Basemap #下面的例子应该是比较综合的 data = pd.read_csv(‘E:\\Haiti.csv‘) #print data #下面处理一下数据,下面的为日期,纬度、经度 #print data[[‘INCIDENT DATE‘,‘LATITUDE‘,‘LONGITUDE‘]][:10] #print data[‘CATEGORY‘][:6] #这些代表消息的类型 #数据中很有可能有异常值、缺失值,下面看一下 #print data.describe() #清除错误信息并移除缺失分类信息是“一件简单的事情” data = data[(data.LATITUDE > 18) & (data.LATITUDE < 20) & (data.LONGITUDE > -75) & (data.LONGITUDE < -70) & data.CATEGORY.notnull()] #我们想根据分类对数据做一些分析或者图形化工作,但是各个分类字段中可能含有多个分类。此外,各个分类信息 #不仅有一个编码,还有一个英语(法语)名称。因此需要对数据进行规整化处理。下面编写两(三)个 #函数,一个用于获取所有分类的列表,一个用于将各个分类信息拆分为编码和英语明名称 #sptrip 是删除空白字符,‘\n‘等;注意作者这种隐式循环写法 def to_cat_list(catstr): stripped = (x.strip() for x in catstr.split(‘,‘)) return [x for x in stripped if x] def get_all_categoties(cat_series): cat_sets = (set(to_cat_list(x)) for x in cat_series) return sorted(set.union(*cat_sets)) def get_english(cat): code,names = cat.split(‘.‘) if ‘|‘ in names: names = names.split(‘|‘)[1] return code,names.strip() #下面进行一下ceshi #print get_english(‘2.Urgences logistiques | Vital Lines‘) #接下来做了一个将编码跟名称映射起来的字典,这是因为我们等会要用编码进行分析。 #下面将所有组合弄出来 all_cats = get_all_categoties(data.CATEGORY) #print data.CATEGORY[:10] #print all_cats #生成器表达式 #生成字典 english_mapping = dict(get_english(x) for x in all_cats) #print english_mapping[‘2a‘] #print english_mapping[‘6c‘] #根据分类选取记录的方式有很多,其中之一就是添加指标(或者哑变量)列,每个分类一列。 #为此,首先抽取出唯一的分类编码,并构造一个全零DataFrame(列为分类编码,索引跟data的索引一样) def get_code(seq): return [x.split(‘.‘)[0] for x in seq if x] #下面是将所有的key取出来 all_codes = get_code(all_cats) #print all_codes code_index = pd.Index(np.unique(all_codes)) #print code_index dummy_frame = DataFrame(np.zeros((len(data),len(code_index))),index = data.index,columns = code_index) #print len(data) #print dummy_frame.ix[:,:6] #下面将各行中适当的项设置为1,然后再与data进行连接: for row,cat in zip(data.index,data.CATEGORY): codes = get_code(to_cat_list(cat)) dummy_frame.ix[row,codes] = 1 #添加前缀,并且合并一下 data = data.join(dummy_frame.add_prefix(‘category_‘)) #print data #接下来开始画图吧,我们希望把数据绘制在海地的地图上。basemap数据集是matplotloib的一个插件 #使得能够用Python在地图上绘制2D数据。basemap提供了许多不同的地球投影以及一种将地球上的经纬度 #坐标投影转换为二维matplotlib图的方式。 #“经过一遍又一遍的尝试”,作者编写了下面的函数,绘制出一张简单的黑白地图。 def basic_haiti_map(ax = None,lllat = 17.25,urlat = 20.25,lllon = -75,urlon = -71): #创建极球面投影的Basemap实例。 m = Basemap(ax = ax,projection = ‘stere‘, lon_0 = (urlon + lllon) / 2, lat_0 = (urlat + lllat) / 2, llcrnrlat = lllat,urcrnrlat = urlat, llcrnrlon = lllon,urcrnrlon = urlon, resolution = ‘f‘ )
由于window下安装geos不成功,这部分等ubuntu装好了再接着写。
4、Python图形化工具生态系统
介绍几个其他的绘图工具。
Chaco
特点:静态图 + 交互图形,非常适合用复杂的图形化方法表示数据的内部关系。对交互支持的好的多,交互式GUI是个不错选择。
mayavi
这是一个基于开源C++图形库VTK的3D图形工具包。可以集成到Ipython交互使用。
其他库
其他库或者应用还有:PyQwt、Veusz、gnuplotpy、biggles等,大部库都在向基于Web的技术发展,并逐渐远离桌面图形技术。
图形化工具的未来
基于Web技术(如Javascript)的图形化是必然的发展趋势,现在已经有不少了,higncharts等。