matplotlib bar函数重新封装

参考：　　https://blog.csdn.net/jenyzhang/article/details/52047557

　　　　　https://blog.csdn.net/liangzuojiayi/article/details/78187704

需求：

封装一个带分组功能的bar绘制函数

绘制效果：

代码：

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

 class Bar(object):
     r'''
     请使用with打开绘图
     '''
     def __init__(self):
         self.colors=[
             "#0ed4e8",
             "#318bae",
             "#6b914c",
             "#8492c3",
             "#495aa2",
             "#6841ac",
             "#aa6547",
             "#aaa350",
         ]

     def __call__(self,quants=None,labels=None,**kwargs):
         self.draw(quants,labels,**kwargs)
     def __enter__(self):
         plt.figure(figsize=(10,10))
         return self
     def __exit__(self, exc_type, exc_val, exc_tb):
         plt.show()
         plt.close()
     def draw(self,quants=None,labels=None,**kwargs):
         if not quants:
             print('数据为空')
             return
         if type(quants[0]) == int or type(quants[0]) == float:
             num_label = 1
             isInt = True
         else:
             num_label = len(quants[0])
             isInt = False
         width = 0.4
         ind = np.linspace(0.5, 9.5, len(quants))
         X=ind - width / 2
         xlabels_pos=X+(num_label-1)*width / 2
         if isInt:
             plt.bar(X, quants, width, color='green')
         else:
             quants=np.array(quants)
             if type(kwargs.get('classlabels', '')) == list:
                 classlabels = kwargs.get('classlabels', '')
                 plt.legend(loc='upper right')
             else:
                 classlabels = ['']*len(quants[0])
             for i in range(len(quants[0])):
                 plt.bar(X+width*i, quants[:,i], alpha=0.9, width=width, facecolor=self.colors[i], edgecolor='white', label=classlabels[i], lw=1)
         # labels
         plt.xlabel(kwargs.get('xlabel', 'x'))
         plt.ylabel(kwargs.get('ylabel', 'y'))
         if not labels:
             labels = list(range(1, len(quants) + 1))
         plt.xticks(xlabels_pos, labels)
         # title
         plt.title(kwargs.get('title', ''))
         plt.grid(linestyle='-.')

 if __name__=="__main__":
     labels = ['USA', 'China', 'India', 'Japan', 'Germany', 'Russia', 'Brazil', 'UK', 'France', 'Italy']
     quants = [15094025.0, 11299967.0, 4457784.0, 4440376.0, 3099080.0, 2383402.0, 2293954.0, 2260803.0, 2217900.0,
               1846950.0]
     labels3 = ['USA', 'China', 'India']
     quants3= [[15094025.0, 11299967.0, 4457784.0], [4440376.0, 3099080.0, 2383402.0], [2293954.0, 2260803.0, 2217900.0]]
     with Bar() as bar:
         plt.subplot(2,1,1)
         bar(quants3,labels3,xlabel="xlabel3",ylabel='ylabel3',title='title3',classlabels=["A",'B','C'])
         plt.subplot(2, 1, 2)
         bar(quants, labels, xlabel="xlabel", ylabel='ylabel', title='title',)