Python matplotlib(2D绘图库)

文章目录

        • matplotlib简介
        • 坐标轴相关设置
          • plt.axis():坐标轴设置
          • plt.xlim() 和 plt.ylim():x和y轴数值范围
          • plt.xlabel()和plt.ylabel:x轴及y轴的标签
          • plt.xticks()和plt.yticks():x轴及y轴刻度
        • plt.figure()的使用
          • subplot创建单个子图
          • subplots创建多个子图
          • add_subplot与add_axes新增子图或区域
        • plt.legend:画图例
        • plt.plot:曲线图、折线图
        • plt.scatter:散点图

matplotlib简介

matplotlib是python最著名的绘图库,它提供了一整套和matlab相似的命令API,十分适合交互式地进行制图。而且也可以方便地将它作为绘图控件,嵌入GUI应用程序中。它的文档相当完备,并且Gallery页面 中有上百幅缩略图,打开之后都有源程序。因此如果你需要绘制某种类型的图,只需要在这个页面中浏览/复制/粘贴一下,基本上都能搞定。

在Linux下比较著名的数据图工具还有gnuplot,这个是免费的,Python有一个包可以调用gnuplot,但是语法比较不习惯,而且画图质量不高。而Matplotlib则比较强:Matlab的语法、python语言、latex的画图质量(还可以使用内嵌的latex引擎绘制的数学公式)。

Matplotlib 图像各个部分
(1)Figure:整个图像为一个Figure对象。一个Figure对象可以包含一个或者多个Axes子图。
Python matplotlib(2D绘图库)_第1张图片
(2)Axes:每个Axes(ax)对象都是一个拥有自己坐标系统的绘图区域。子图Axes可以用subplot()快速绘制。
各个对象关系可以梳理成以下内容:
图像中所有对象均来自于Artist的基类。
Python matplotlib(2D绘图库)_第2张图片

当前的图表和子图可以使用plt.gcf()和plt.gca()获得,分别表示Get Current Figure和Get Current Axes。

matplotlib.pyplot
matplotlib.pyplot(下面简写成plt)是一些命令行风格函数的集合,使matplotlib以类似于MATLAB的方式工作。每个plt函数对一幅图片(figure)做一些改动:比如创建新图片,在图片创建一个新的作图区域(plotting area),在一个作图区域内画直线,给图添加标签(label)等。plt是有状态的,亦即它会保存当前图片和作图区域的状态,新的作图函数会作用在当前图片的状态基础之上。

plt可在图片上做的一些修改如下:
(1)plt.axis:坐标轴

plt.axis([0,6,0,20])
axis()函数接受形如[xmin,xmax,ymin,ymax]的参数,指定了X,Y轴坐标的范围。

(2)Tick:刻度线,Tick Label:刻度注释 plt.xticks(np.linspace(-1, 1, 5)) #x坐标为从-1到1平均分成五个点:-1,-0.5,0.5,1
(3)plt.xlabel,plt.ylabel:坐标轴标注
(4)plt.title:图像标题
plt.title(s, *args, **kwargs)
为当前图表添加标题,重要参数如下:
s:标题内容
fontdict:标题样式,是一个字典
loc:标题位置,可选 ‘centet’, ‘left’, ‘right’

Python matplotlib(2D绘图库)_第3张图片

(5)plt.legend():显示图例

坐标轴相关设置

plt.axis():坐标轴设置

函数
plt.axis(*v, **kwargs)

主要用于设置坐标轴的属性,返回值为当前的坐标轴范围 [xmin, xmax, ymin, ymax],几种调用方式如下:

调用方式 说明
axis() 返回当前的坐标轴范围 [xmin, xmax, ymin, ymax]
axis(v) 其中 v 为 [xmin, xmax, ymin, ymax]用来设置坐标轴范围
axis(‘off’) 不显示坐标轴和坐标轴名称
axis(‘equal’) x、y轴对应长度所表示的数值相同,提高某一个轴的范围以保持图表大小及比例不变
axis(‘scaled’) x、y 轴对应长度所表达的数值相同,并保持两个轴的数值范围不变,因此需要对图表比例进行调整(缩放)
axis(‘tight’) 在显示所有数据的前提下,尽量减少两个轴的数值范围,并尽量让数据居中
axis(‘image’) 将图表比例按照图片的比例进行调整(缩放)
axis(‘square’) 将图表变成正方形,并确保 (xmax-xmin) 与 (ymax-ymin) 相同
axis(‘auto’) 恢复自动范围
plt.xlim() 和 plt.ylim():x和y轴数值范围

设置并返回 x 轴和 y 轴的数值范围,以 xlim() 为例说明调用方式:

调用方式 说明
xlim() 返回 xmin, xmax
xlim(xmin, xmax) 或 xlim((xmin, xmax)) 设置 x 轴的最大、最小值
xlim(xmax = n) 和 xlim(xmin = n) 设置 x 轴的最大或最小值
plt.xlabel()和plt.ylabel:x轴及y轴的标签

函数
plt.xlabel(label, fontdict=None, labelpad=None, **kwargs)

plt.xticks()和plt.yticks():x轴及y轴刻度

plt.xticks(locs, [labels], **kwargs) # Set locations and labels
locs表示位置,labels决定这些位置上的标签,labels的默认值为和locs相同。

# --coding:utf-8--
import matplotlib.pyplot as plt

数据设置

x1 = [0, 5000, 10000, 15000, 20000, 25000, 30000, 35000, 40000, 45000, 50000, 55000];
y1 = [0, 223, 488, 673, 870, 1027, 1193, 1407, 1609, 1791, 2113, 2388];

x2 = [0, 5000, 10000, 15000, 20000, 25000, 30000, 35000, 40000, 45000, 50000, 55000];
y2 = [0, 214, 445, 627, 800, 956, 1090, 1281, 1489, 1625, 1896, 2151];

设置输出的图片大小

figsize = 7,5
figure, ax = plt.subplots(figsize=figsize)

在同一幅图片上画两条折线

A, = plt.plot(x1, y1, ‘-r’, label=‘A’, linewidth=5.0)
B, = plt.plot(x2, y2, ‘b-.’, label=‘B’, linewidth=5.0)

设置图例并且设置图例的字体及大小

font1 = {‘family’: ‘Times New Roman’,
‘weight’: ‘normal’,
‘size’: 23,
}
legend = plt.legend(handles=[A, B], prop=font1)

设置坐标刻度值的大小以及刻度值的字体

plt.tick_params(labelsize=23)
labels = ax.get_xticklabels() + ax.get_yticklabels()
[label.set_fontname(‘Times New Roman’) for label in labels]

设置横纵坐标的名称以及对应字体格式

font2 = {‘family’: ‘Times New Roman’,
‘weight’: ‘normal’,
‘size’: 30,
}
plt.xlabel(‘round’, font2)
plt.ylabel(‘value’, font2)

将文件保存至文件中并且画出图

plt.savefig(‘figure.eps’)
plt.show()

    Python matplotlib(2D绘图库)_第4张图片

    plt.figure()的使用

    函数

    plt.figure(num=None, figsize=None, dpi=None, facecolor=None, edgecolor=None, frameon=True)

    重要参数

    • num:图像编号或名称,数字为编号 ,字符串为名称
    • figsize:指定figure的宽和高,单位为英寸;
    • dpi:指定绘图对象的分辨率,即每英寸多少个像素,缺省值为80 1英寸等于2.5cm,A4纸是 21*30cm的纸张
    • facecolor:背景颜色
    • edgecolor:边框颜色
    • frameon:是否显示边框

    例子1

    import matplotlib.pyplot as plt
    创建自定义图像
    fig=plt.figure(figsize=(4,3),facecolor='blue')
    plt.show()
    
       
       
       
       
      subplot创建单个子图

      matplotlib下, 一个 Figure 对象可以包含多个子图(Axes), 可以使用 subplot() 快速绘制, 其调用形式如下 :
      subplot(nrows,ncols,sharex,sharey,subplot_kw,**fig_kw)

      • 图表的整个绘图区域被分成 numRows 行和 numCols 列
      • 然后按照从左到右,从上到下的顺序对每个子区域进行编号,左上的子区域的编号为1
      • plotNum 参数指定创建的Axes对象所在的区域

      例如,如果 numRows = 2, numCols = 2, 那整个绘制图表样式为 2X2 的图片区域, 用坐标表示为
      (1, 1), (1, 2)
      (2, 1), (2, 2)
      这时, 当 plotNum = 2时,表示的坐标为(1, 2), 即第一行第二列的子图

      (1)如果 numRows, numCols 和 plotNum 这三个数都小于 10 的话, 可以把它们缩写为一个整数, 例如 subplot(222) 和 subplot(2,2,2) 是相同的。
      (2)subplot在 plotNum 指定的区域中创建一个轴对象. 如果新创建的轴和之前创建的轴重叠的话,之前的轴将被删除。

      示例1
      规则划分成2*2的

      import matplotlib.pyplot as plt
      

      plt.figure()

      绘制第一个图

      plt.subplot(2, 2, 1)
      plt.plot([0, 1], [0, 1])

      绘制第二个图

      plt.subplot(2, 2, 2)
      plt.plot([0, 1], [0, 1])

      绘制第三个图

      plt.subplot(2, 2, 3)
      plt.plot([0, 1], [0, 1])

      绘制第四个图

      plt.subplot(2, 2, 4)
      plt.plot([0, 1], [0, 1])
      a,b,c,d=plt.axis()
      print(“a:”,a," b:",b," c:"," d:",d) #a: -0.05 b: 1.05 c: d: 1.05
      plt.show()

        Python matplotlib(2D绘图库)_第5张图片

        不规则的

        import matplotlib.pyplot as plt
        

        plt.figure()

        绘制第一个图

        plt.subplot(2, 1, 1)
        plt.plot([0, 1], [0, 1])

        绘制第二个图

        plt.subplot(2, 3, 4)
        plt.plot([0, 1], [0, 1])

        绘制第三个图

        plt.subplot(2, 3, 5)
        plt.plot([0, 1], [0, 1])

        绘制第四个图

        plt.subplot(2, 3, 6)
        plt.plot([0, 1], [0, 1])
        plt.show()

          Python matplotlib(2D绘图库)_第6张图片
          import matplotlib.pyplot as plt
          import numpy as np
          

          def f(t):
          return np.exp(-t) * np.cos(2 * np.pi * t)

          t1 = np.arange(0, 5, 0.1)
          t2 = np.arange(0, 5, 0.02)

          plt.figure(12)#12代表第几个figure(左上角)

          分成2x2,占用第一个,即第一行第一列的子图

          plt.subplot(221)
          plt.plot(t1, f(t1), ‘bo’, t2, f(t2), ‘r–’)

          分成2x2,占用第二个,即第一行第二列的子图

          plt.subplot(222)
          plt.plot(t2, np.cos(2 * np.pi * t2), ‘r–’)

          将整个Figure,分成2x1,占用第二个,即第二行

          plt.subplot(212)
          plt.plot([1, 2, 3, 4], [1, 4, 9, 16])

          plt.show()

            Python matplotlib(2D绘图库)_第7张图片
            subplots创建多个子图

            subplots参数与subplots相似。两者都可以规划figure划分为n个子图,但每条subplot命令只会创建一个子图,而一条subplots就可以将所有子图创建好。subplots返回一个 Figure实例fig 和一个 AxesSubplot实例ax 。
            fig, ax = plt.subplots()

            import numpy as np
            import matplotlib.pyplot as plt
            

            x = np.arange(0, 100)

            划分子图

            fig, axes = plt.subplots(2, 2) #2行2列
            ax1 = axes[0, 0]
            ax2 = axes[0, 1]
            ax3 = axes[1, 0]
            ax4 = axes[1, 1]

            作图1

            ax1.plot(x, x)

            作图2

            ax2.plot(x, -x)

            作图3

            ax3.plot(x, x ** 2)
            ax3.grid(color=‘r’, linestyle=’–’, linewidth=1, alpha=0.3)

            作图4

            ax4.plot(x, -2xx+2*x)
            plt.savefig(“test.jpg”)
            plt.show()

              Python matplotlib(2D绘图库)_第8张图片
              add_subplot与add_axes新增子图或区域
              import matplotlib.pyplot as plt
              import numpy as np
              

              def f(t):
              return np.exp(-t) * np.cos(2 * np.pi * t)

              t1 = np.arange(0, 5, 0.1)
              t2 = np.arange(0, 5, 0.02)
              #使用add方式
              fig=plt.figure(12)#12代表第几个figure(左上角)

              分成2x2,占用第一个,即第一行第一列的子图

              ax1=fig.add_subplot(221)
              ax1.plot(t1, f(t1), ‘bo’, t2, f(t2), ‘r–’)

              分成2x2,占用第二个,即第一行第二列的子图

              ax2=fig.add_subplot(222)
              ax2.plot(t2, np.cos(2 * np.pi * t2), ‘r–’)

              将整个Figure,分成2x1,占用第二个,即第二行

              ax3=fig.add_subplot(212)
              ax3.plot([1, 2, 3, 4], [1, 4, 9, 16])

              plt.show()

                Python matplotlib(2D绘图库)_第9张图片

                plt.legend:画图例

                1.使用pyplot的方式
                legend语法参数如下: matplotlib.pyplot.legend(*args,**kwargs)
                一些重要参数如下:
                (1)设置图例位置
                使用loc参数
                plt.legend(loc=‘lower left’)
                有下面一些组合:
                ‘best’ : 0, (only implemented for axes legends)(自适应方式) ‘upper right’ : 1, ‘upper left’ : 2, ‘lower left’ : 3, ‘lower right’ : 4, ‘right’ : 5, ‘center left’ : 6, ‘center right’ : 7, ‘lower center’ : 8, ‘upper center’ : 9, ‘center’ : 10

                (2)设置图例字体
                #设置字体大小
                fontsize: int or float or {‘xx-small’, ‘x-small’, ‘small’, ‘medium’, ‘large’, ‘x-large’, ‘xx-large’}

                (3)设置图例边框及背景
                plt.legend(loc=‘best’,frameon=False) #去掉图例边框
                plt.legend(loc=‘best’,edgecolor=‘blue’) #设置图例边框颜色
                plt.legend(loc=‘best’,facecolor=‘blue’) #设置图例背景颜色,若无边框,参数无效

                (4)设置图例标题
                plt.legend(loc=‘best’,title=‘figure 1 legend’)

                2.使用面向对象的方式
                (1)获取并设置legend图例

                plt.legend(loc=0, numpoints=1)  
                leg = plt.gca().get_legend() #或leg=ax.get_legend()  
                ltext = leg.get_texts()  
                plt.setp(ltext, fontsize=12,fontweight='bold') 
                
                   
                   
                   
                   

                  (2)设置图例

                  legend = ax.legend(loc='upper center', shadow=True, fontsize='x-large')  
                  legend.get_frame().set_facecolor('red')   #设置图例legend背景为红色  
                  frame = legend.get_frame()  
                  frame.set_alpha(1)  
                  frame.set_facecolor('none') #设置图例legend背景透明  
                  
                     
                     
                     
                     

                    (3)移除图例
                    ax1.legend_.remove() ##移除子图ax1中的图例

                    示例1:显示多图例legend

                    import matplotlib.pyplot as plt
                    import numpy as np
                    

                    x = np.random.uniform(-1, 1, 4)
                    y = np.random.uniform(-1, 1, 4)
                    p1, = plt.plot([1, 2, 3])
                    p2, = plt.plot([3, 2, 1])
                    l1 = plt.legend([p2, p1], [“line 2”, “line 1”], loc=‘upper left’)

                    p3 = plt.scatter(x[0:2], y[0:2], marker=‘D’, color=‘r’)
                    p4 = plt.scatter(x[2:], y[2:], marker=‘D’, color=‘g’)

                    This removes l1 from the axes.

                    plt.legend([p3, p4], [‘label’, ‘label1’], loc=‘lower right’, scatterpoints=1)

                    Add l1 as a separate artist to the axes

                    plt.gca().add_artist(l1)
                    plt.show()

                      示例2:

                      import matplotlib.pyplot as plt
                      #蓝线
                      line1, = plt.plot([1,2,3], label="Line 1", linestyle='--')
                      #橘黄色的线
                      line2, = plt.plot([3,2,1], label="Line 2", linewidth=4)
                      # 为第一个线条创建图例
                      first_legend = plt.legend(handles=[line1], loc=1)
                      # plt.gca(): 返回当前axes(matplotlib.axes.Axes)
                      # plt.gcf(): 返回当前figure(matplotlib.figure.Figure)
                      # plt.clf(): 清理当前figure
                      # plt.cla(): 清理当前axes
                      # plt.close(): 一副figure知道显示的调用close()时才会释放她所占用的资源;
                      # 手动将图例添加到当前轴域
                      ax = plt.gca().add_artist(first_legend)
                      # 为第二个线条创建另一个图例
                      plt.legend(handles=[line2], loc=4)
                      plt.show()
                      
                         
                         
                         
                         

                        plt.plot:曲线图、折线图

                        plt.plot画图时可以设定线条参数。包括:颜色、线型、标记风格。

                        只提供一个列表或数组(Y向量)-缺省x
                        如果只提供给plot()函数一个列表或数组,matplotlib会认为这是一串Y值(Y向量),并且自动生成X值(X向量),x的默认值是:range(len(y))

                        import matplotlib.pyplot as plt
                        plt.plot([1,2,3,4])
                        plt.ylabel('some numbers')
                        plt.show()
                        
                           
                           
                           
                           
                          Python matplotlib(2D绘图库)_第10张图片

                          提供X,Y向量
                          也可以给plt.plot()函数传递多个序列(元组或列表),每两个序列是一个X,Y向量对,在图中构成一条折线。

                          如果要显示的制定X轴的坐标,可以像如下一样:
                          plt.plot([1,2,3,4],[1,4,9,16])
                          形成点(1,1),(2,4),(3,9),(4,16)

                          控制线形(marker)和颜色

                          (1)线形,默认是实线solid-
                              ``'-'``             solid line style
                              ``'--'``            dashed line style
                              ``'-.'``            dash-dot line style
                              ``':'``             dotted line style
                              
                          (2)点的形状:marker,默认为point .
                              ``'.'``             point marker
                              ``','``             pixel marker
                              ``'o'``             circle marker
                              ``'v'``             triangle_down marker
                              ``'^'``             triangle_up marker
                              ``'<'``             triangle_left marker
                              ``'>'``             triangle_right marker
                              ``'1'``             tri_down marker
                              ``'2'``             tri_up marker
                              ``'3'``             tri_left marker
                              ``'4'``             tri_right marker
                              ``'s'``             square marker
                              ``'p'``             pentagon marker
                              ``'*'``             star marker
                              ``'h'``             hexagon1 marker
                              ``'H'``             hexagon2 marker
                              ``'+'``             plus marker
                              ``'x'``             x marker
                              ``'D'``             diamond marker
                              ``'d'``             thin_diamond marker
                              ``'|'``             vline marker
                              ``'_'``             hline marker
                              
                          (3)颜色对应如下,默认为蓝色b
                              'b'         blue
                              'g'         green
                              'r'         red
                              'c'         cyan
                              'm'         magenta
                              'y'         yellow
                              'k'         black
                              'w'         white
                          
                          

                          示例:
                          为了区分同一个图里的多条曲线,可以为每个X,Y向量对指定一个参数来标明该曲线的表现形式,默认的参数是’b-’,亦即蓝色的直线,如果想用红色的圆点来表示这条曲线,可以:

                          import matplotlib.pyplot as plt
                          plt.plot([1,2,3,4],[1,4,9,16],'ro')
                          

                            matplotlib不仅仅可以使用序列(列表和元组)作为参数,还可以使用numpy数组。实际上,所有的序列都被内在的转化为numpy数组。同时plot函数还可以同时绘制多条线。

                            import numpy as np
                            import matplotlib.pyplot as plt
                            t=np.arange(0.,5.,0.2)
                            plt.plot(t,t,'r--',t,t**2,'bs',t,t**3,'g^')#绘制三条线,t**2表示t的平方,即每个元素的都平方
                            plt.show()
                            
                               
                               
                               
                               
                              Python matplotlib(2D绘图库)_第11张图片

                              线条、marker、颜色可以无顺序任意组合

                              import numpy as np
                              import matplotlib.pyplot as plt
                              t=np.arange(0.,5.,0.2)
                              plt.plot(t,t,'r--*',t,t**2,'s--b',t,t**3,':g')#绘制三条线,t**2表示t的平方,即每个元素的都平方
                              plt.axis([0,6,0,20])
                              #axis()函数接受形如[xmin,xmax,ymin,ymax]的参数,指定了X,Y轴坐标的范围。
                              plt.show()
                              
                                 
                                 
                                 
                                 
                                Python matplotlib(2D绘图库)_第12张图片

                                plt.scatter:散点图

                                语法:plt.scatter(x, y, s, c ,marker, alpha,…)

                                • x,y: x轴与y轴的数据,都是向量,而且必须长度相等
                                • s: 点的面积。想让所有点大小一样就填一个数字,想不一样就填一个向量
                                • c: 点的颜色。可填成RGB 三元数或颜色名称或由 RGB 三元数组成的三列矩阵或颜色向量
                                选项 说明 对应的 RGB 三元数
                                ‘red’ 或 ‘r’ 红色 [1 0 0]
                                ‘green’ 或 ‘g’ 绿色 [0 1 0]
                                ‘blue’ 或 ‘b’ 蓝色 [0 0 1]
                                ‘yellow’ 或 ‘y’ 黄色 [1 1 0]
                                ‘magenta’ 或 ‘m’ 品红色 [1 0 1]
                                ‘cyan’ 或 ‘c’ 青蓝色 [0 1 1]
                                ‘white’ 或 ‘w’ 白色 [1 1 1]
                                ‘black’ 或 ‘k’ 黑色 [0 0 0]
                                #不同颜色的点
                                cValue = ['r','y','g','b','r','y','g','b','r']
                                plt.scatter(x,y,c=cValue)
                                
                                   
                                   
                                   
                                   
                                • 1
                                • 2
                                • 3
                                • marker: 点的形状
                                • alpha: 透明度。[0,1]:1不透明,0透明
                                • edgecolors:轮廓颜色。和c类似,参数也相同
                                • linewidths:线宽。标记边缘的宽度,默认是’face’

                                注意事项
                                color、marker等不能同时作为一个参数,plt.scatter(x1, y1, ‘bo’, s=5)不合法。

                                示例1:

                                from numpy import *  
                                import matplotlib  
                                import matplotlib.pyplot as plt  
                                import numpy as np
                                x = random.rand(50,30)      #创建一个50行30列的多维数组(ndarray)
                                #basic  
                                f1 = plt.figure(1)          #创建显示图形输出的窗口对象
                                plt.subplot(211)            #创建子坐标系
                                #x[:,1]获取第二列作为一维数组,x[:,0]获取第一列作为一维数组
                                plt.scatter(x[:,1],x[:,0])  #画散点图
                                

                                with label

                                plt.subplot(212) #c重新创建坐标系

                                #list(ones(20)) 创建1行20列的全为1列表
                                #list(2ones(15)) 创建1行15列全为2的列表
                                #list(3
                                ones(15) 创建1行15列全为3的列表

                                label = list(ones(20))+list(2ones(15))+list(3ones(15)) #将列表合并到一起,共50列
                                label = array(label) #将列表转为数组

                                #15.0*label 将数组的每个值都乘以15.0
                                #x[:,1] 将x的第2列50行转为1行50列
                                #x[:,0] 将x的第1列50行转为1行50列

                                #x轴和y轴均50个点,两个Label都是1行50列的数组
                                #从第一个点到第20个点的样式相同,从第21到第35个点相同,从第36到第50个点相同
                                plt.scatter(x[:,1],x[:,0],15.0label,15.0label)

                                with legend

                                f2 = plt.figure(2) #创建显示图形输出的窗口对象
                                idx_1 = np.where(label==1) #找label中为1的位置

                                #画图 marker标识散点图样式 color标识颜色 label表示图例的解释 s表示散点的大小
                                p1 = plt.scatter(x[idx_1,1], x[idx_1,0], marker = ‘x’, color = ‘m’, label=‘1’, s = 30)
                                idx_2 = np.where(label==2) #找label中为2的位置

                                p2 = plt.scatter(x[idx_2,1], x[idx_2,0], marker = ‘+’, color = ‘c’, label=‘2’, s = 50)
                                idx_3 = np.where(label==3) #找label中为3的位置

                                p3 = plt.scatter(x[idx_3,1], x[idx_3,0], marker = ‘o’, color = ‘r’, label=‘3’, s = 15)
                                plt.legend(loc = ‘upper right’) #图例的位置

                                plt.show()

                                  Python matplotlib(2D绘图库)_第13张图片

                                  Python matplotlib(2D绘图库)2

                                  你可能感兴趣的:(matplotlib)