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

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()