python 绘制饼状图

前言

测试执行结果发送报告时，分析图标比文案描述表达效果更好，来看下效果吧

 

 

 

代码实现

import matplotlib.pyplot as plt

def pie(x,labels,title):
    plt.figure(figsize=(7,5))
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.pie(x,labels=labels,radius=0.9,autopct='%0.1f%%')
    plt.title(title)
    plt.show()
if __name__ == '__main__':
    labels = ['UI自动化：10条', '接口自动化：2条', 'codeReview：3条', '静态扫描：1条', '单元测试：4条']
    x = [10, 2, 3, 1, 4]
    title = '不合格类型占比分布图'
    pie(x,labels,title)

以上操作可以在控制台输出饼图，但是需要展示在前端或者邮件中，就需要保存成文件格式，代码如下

import matplotlib.pyplot as plt
import time
import base64

def pie(x,labels,title):
    plt.figure(figsize=(7,5))
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.pie(x,labels=labels,radius=0.9,autopct='%0.1f%%')
    plt.title(title)
    #控制台展示
    # plt.show()
    #保存为文件
    image_path = "D:\\develop\\vega\\source\\pie.jpeg"
    plt.savefig(image_path)
    time.sleep(3)

    #jpeg转为bytes
    with open(image_path,'rb') as f:
        s = f.read()
        res = base64.b64encode(s)
        return str(res.decode())


if __name__ == '__main__':
    labels = ['UI自动化：10条', '接口自动化：2条', 'codeReview：3条', '静态扫描：1条', '单元测试：4条']
    x = [10, 2, 3, 1, 4]
    title = '不合格类型占比分布图'
    image_bytes = pie(x,labels,title)
    print(image_bytes)

 

在django templates中展示

#report.html
...
<img src="data:image/jpeg;base64,{{ image_bytes }}" height="100" width="100"  />
...

 

饼状图绘图原理

Python中绘制饼状图需用matplotlib.pyplot中的pie函数，该函数的基本语法为：

pie(x, [explode], [labels], [colors], [autopct], [pctdistance],  [labeldistance], [startangle], [radius], [textprops], **kwargs)

参数说明：
  
x：数组，绘制饼状图的数据。
  
[explode]：默认值为None的可选参数。若非None，则是和x相同长度的数组，用来指定每部分的离心偏移量。
  
[labels]：列表，指定每个饼块的名称，默认值None，为可选参数。
  
[colors]：特定字符或数组，指定饼图的颜色，默认值None，为可选参数。
  
[autopct]：特定字符，指定饼图中数据标签的显示方式，默认值None，为可选参数。
  
[pctdistance]：浮点数，指定显示比例距离圆心的距离。默认值0.6，为可选参数。
  
[labeldistance]：浮点数，指定每个扇形对应标签与圆心的距离，默认值1.1，为可选参数。
  
[startangle]：浮点数，指定从x轴逆时针旋转饼图的开始角度，默认值None，为可选参数。
  
[radius]：浮点数，指定饼图的半径，默认值1，为可选参数。
  
[textprops]：字典，设置文本对象的字典参数，默认值None，为可选参数。
  
**kwargs：不定长关键字参数，用字典形式设置条形图的其它参数。

参考文档：https://blog.csdn.net/qq_32532663/article/details/113631551