matplotlib绘制树形图之基本配置——万能模板案例

对于树形图的绘制，首先我们需要安装一个第三方库，才可以进行图像的绘制

!pip install squarify
 

  

   

    

     

    
    

     

      import pandas as pd
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      import squarify
     
    
   

    

     

    
    

     

      import matplotlib.pyplot as plt
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      plt.rcParams['font.sans-serif']=['SimHei']     #显示中文
     
    
   

    

     

    
    

     

      plt.rcParams['axes.unicode_minus']=False       #正常显示负号
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      data = pd.read_csv(r'../data/orders.txt')
     
    
   

    

     

    
    

     

      # sql_num = "SELECT region, ROUND(SUM(sales/10000),2) FROM orders WHERE dt=2019 GROUP BY region"
     
    
   

    

     

    
    

     

      data_region = np.round( data.groupby('CUST_REGION')['PRICE'].sum()/10000 )
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      plt.figure(figsize=(6,6))
     
    
   

    

     

    
    

     

      colors = ['steelblue','red','indianred','green','yellow','orange']   #设置颜色数据
     
    
   

    

     

    
    

     

      plot=squarify.plot(
     
    
   

    

     

    
    

     

          sizes=data_region.values,            #指定绘图数据
     
    
   

    

     

    
    

     

          label=data_region.index,            #标签
     
    
   

    

     

    
    

     

          color=colors,        #指定自定义颜色
     
    
   

    

     

    
    

     

          alpha=0.6,           #指定透明度
     
    
   

    

     

    
    

     

          value=data_region.values,            #添加数值标签
     
    
   

    

     

    
    

     

          edgecolor='w',   #设置边界框白色
     
    
   

    

     

    
    

     

          linewidth=8          #设置边框宽度为3
     
    
   

    

     

    
    

     

      )
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      plt.rc('font',size=15)      #设置标签大小
     
    
   

    

     

    
    

     

      plot.set_title('2019年企业销售额情况',fontdict={'fontsize':15})    #设置标题及大小
     
    
   

    

     

    
    

     

      plt.axis('off')    #去除坐标轴
     
    
   

    

     

    
    

     

      plt.tick_params(top='off',right='off')    #去除上边框和右边框刻度
     
    
   

    

     

    
    

     

      plt.show()
     
    
  
 

 每文一语

简单不失特色！

文章来源: wxw-123.blog.csdn.net，作者：王小王-123，版权归原作者所有，如需转载，请联系作者。

原文链接：wxw-123.blog.csdn.net/article/details/124063239