使用matplotlib画图的简单封装

ゝ一纸荒年。 2022-06-17 08:46 271阅读 0赞

import matplotlib.pyplot as plt
    from mpl_toolkits.mplot3d import Axes3D
    import numpy as np
    '''
    设置绘图对象
    '''
    def setFigure(size):
        plt.figure(figsize=size)
    
    '''
    设置标题
    '''
    def setTitle(title):
        plt.title(title)
    
    '''
    设置X轴描述
    '''
    def setXlabel(xlable):
        plt.xlabel(xlable)
    
    '''
    设置Y轴描述
    '''
    def setYlabel(ylable):
        plt.ylabel(ylable)
    
    '''
    画线性图
    '''
    def plotLineChart(x_axis,y_axis,color,lw,label):
         plt.plot(x_axis, y_axis, color, lw=lw, label=label)
    
    '''
    画柱状图
    edgecolor柱状图的边框颜色 align x轴的ticks对齐方式  
    color 柱状图的颜色 label 图示标题 
    '''
    def plotBarChart(x,y,color,edgecolor,label,align):
        plt.bar(x, y, color, edgecolor, label, align)
    
    '''
    画散点图
    s = size 点的大小 c = color alpha = 点的透明度  
    '''
    def plotScatter(x,y,color,alpha):
        plt.scatter(x, y, s=15, c=color, alpha = alpha)
    
    '''
    画3D图
    cmp plt.cm.jet/ plt.get_cmap('rainbow')
    '''
    def plot3DChart(x,y,z):
        # 设置三维坐标
        fig = plt.figure()
        ax = Axes3D(fig)
        X, Y = np.meshgrid(x, y)  # XY平面的网格数据
        # 画3d图
        ax.plot_surface(X, Y, z, rstride=1, cstride=1, cmap=plt.cm.jet)
    
    '''
    显示每个图的y值  
    ha va 为水平和垂直对齐方式  
    将x/y作为一对（x,y）  
    '''
    def setBarText(x,y):
        data = zip(x,y)
        for x,y in data:
             plt.text(x+0.1,y+0.1,float(y),ha='center',va='bottom')
    
    
    ''' 
    设置legend 的位置参数
        best
    	upper right
    	upper left
    	lower left
    	lower right
    	right
    	center left
    	center right
    	lower center
    	upper center
    	center
    '''
    def setLegend(loc):
         plt.legend(loc=loc)
    
    ''''
    设置X轴的范围 plt.xlim(x.min()*1.1, x.max()*1.1)
    '''
    def setXLim(min,max):
        plt.xlim(min, max)
    '''
    设置Y轴的范围 plt.ylim(c.min()*1.1, s.max()*1.1)
    '''
    def setYLim(min,max):
        plt.ylim(min, max)
    
    #根据相应坐标数 转换成字符坐标
    '''
    设置X坐标的对应表示
    '''
    def setXTicks(x_data,x_label):
        plt.xticks(x_data,x_label)
    
    '''
     设置Y坐标的对应表示
     '''
    def setYTicks(y_data,y_label):
        plt.yticks(y_data,y_label)
    
    '''
    设置坐标轴的位置
    '''
    def setAxisPosition():
        ax = plt.gca()
        #先把右边和上边的边界设置为不可见
        ax.spines['right'].set_color('none')
        ax.spines['top'].set_color('none')
        #然后把下边界和左边界移动到0点
        ax.xaxis.set_ticks_position('bottom')
        ax.spines['bottom'].set_position(('data',0))
        ax.yaxis.set_ticks_position('left')
        ax.spines['left'].set_position(('data',0))
    
    '''
    添加标注
    '''
    def setAnnotate(x0,y0,color):
        plt.scatter(x0,y0,color=color)  #显示一个点
        plt.plot([x0,x0],[0,y0],'b--') #在点到x轴画出垂直线
        #标注方法1
        plt.annotate('y = sin(x)' % y0,xy=(x0,y0),xycoords='data',xytext=(+30,-30),textcoords='offset points',
                 arrowprops=dict(arrowstyle='->',connectionstyle='arc3,rad=-0.2'))
    
    '''
    标注方法2
    '''
    def setText(x0,y0,text):
        plt.text(x0+0.1, y0,text)
    
    '''
    设置坐标轴字体的透明度
    '''
    def setTransparency():
        ax=plt.gca()
        for label in ax.get_xticklabels()+ax.get_yticklabels():
            label.set_fontsize(12)
            label.set_bbox(dict(facecolor='white',edgecolor='none',alpha=0.7))
    
    '''
    设置多个子图区 
    plt.subplot(xyz)
    x表示行
    y表示列
    z表示图表序号位置
    '''
    def setSubPlot(posion):
        plt.subplot(posion)
    
    '''
    显示图表
    '''
    def showChart():
        plt.show()