python批量制作雷达图的实现方法


Posted in Python onJuly 26, 2016

前言

因为工作需要有时候要画雷达图,但是数据好多组怎么办?不能一个一个点excel去画吧,那么可以利用python进行批量制作,得到样式如下:

python批量制作雷达图的实现方法

首先制作一个演示的excel,评分为excel随机数生成:

1 =INT((RAND()+4)*10)/10

加入标签等得到的excel样式如下(部分,共计32行):

python批量制作雷达图的实现方法

那么接下来就是打开python写码了,本文是基于pycharm进行编写

 

wb = load_workbook(filename=r'C:\Users\Administrator\Desktop\数据指标.xlsx') ##读取路径
   ws = wb.get_sheet_by_name("Sheet1") ##读取名字为Sheet1的sheet表
 
   info_id = []
   info_first = []
 
   for row_A in range(2, 32): ## 遍历第2行到32行
     id = ws.cell(row=row_A, column=1).value ## 遍历第2行到32行,第1列
     info_id.append(id)
   for col in range(2, 9): ##读取第1到9列
     first = ws.cell(row=1, column=col).value
     info_first.append(first) ##得到1到8列的标签
 
   info_data = []
   for row_num_BtoU in range(2, len(info_id) + 2): ## 遍历第2行到32行
     row_empty = [] ##建立一个空数组作为临时储存地,每次换行就被清空
     for i in range(2, 9): ## 遍历第2行到32行,第2到9列
       data_excel = ws.cell(row=row_num_BtoU, column=i).value
       if data_excel == None:
         pass
       else:
         row_empty.append(data_excel) ##将单元格信息储存进去
     info_data.append(row_empty)

分步讲解:

读取excel表格:

wb = load_workbook(filename=r'C:\Users\Administrator\Desktop\数据指标.xlsx') ##读取路径
   ws = wb.get_sheet_by_name("Sheet1") ##读取名字为Sheet1的sheet表

需要用到库:

 import xlsxwriter

 from openpyxl import load_workbook

在命令指示符下输入:

 pip install xlsxwriter

等待安装即可,后面的库也是如此:

python批量制作雷达图的实现方法

将第一列ID储存,以及第一行的标签,标签下面的数值分别储存在:

info_id = []
  info_first = []
  info_data = []

读取数据后接下来需要设置写入的格式:

workbook = xlsxwriter.Workbook('C:\\Users\\Administrator\\Desktop\\result.xlsx')
   worksheet = workbook.add_worksheet() # 创建一个工作表对象
   #字体格式
   font = workbook.add_format(
     {'border': 1, 'align': 'center', 'font_size': 11, 'font_name': '微软雅黑'}) ##字体居中,11号,微软雅黑,给一般的信息用的
   #写下第一行第一列的标签
   worksheet.write(0, 0, '商品货号', font)
   ##设置图片的那一列宽度
   worksheet.set_column(0, len(info_first) + 1, 11) # 设定第len(info_first) + 1列的宽度为11

将标签数据等写入新的excel表格中:

#新建一个excel保存结果
   workbook = xlsxwriter.Workbook('C:\\Users\\Administrator\\Desktop\\result.xlsx')
   worksheet = workbook.add_worksheet() # 创建一个工作表对象
   #字体格式
   font = workbook.add_format(
     {'border': 1, 'align': 'center', 'font_size': 11, 'font_name': '微软雅黑'}) ##字体居中,11号,微软雅黑,给一般的信息用的
   #写下第一行第一列的标签
   worksheet.write(0, 0, '商品货号', font)
   ##设置图片的那一列宽度
   worksheet.set_column(0, len(info_first) + 1, 11) # 设定第len(info_first) + 1列的宽度为11
 
   ##写入标签
   for k in range(0,7):
     worksheet.write(0, k + 1, info_first[k], font)
   #写入最后一列标签
   worksheet.write(0, len(info_first) + 1, '雷达图', font)

制作雷达图:

#设置雷达各个顶点的名称
   labels = np.array(info_first)
   #数据个数
   data_len = len(info_first)
   for i in range(0,len(info_id)):
     data = np.array(info_data[i])
 
     angles = np.linspace(0, 2*np.pi, data_len, endpoint=False)
     data = np.concatenate((data, [data[0]])) # 闭合
     angles = np.concatenate((angles, [angles[0]])) # 闭合
 
     fig = plt.figure()
     ax = fig.add_subplot(111, polar=True)# polar参数!!
     ax.plot(angles, data, 'bo-', linewidth=2)# 画线
     ax.fill(angles, data, facecolor='r', alpha=0.25)# 填充
     ax.set_thetagrids(angles * 180/np.pi, labels, fontproperties="SimHei")
     ax.set_title("商品货号:" + str(info_id[i]), va='bottom', fontproperties="SimHei")
     ax.set_rlim(3.8,5)# 设置雷达图的范围
     ax.grid(True)
     plt.savefig("C:\\Users\\Administrator\\Desktop\\result\\商品货号:" + str(info_id[i]) + ".png", dpi=120)

图片太大怎么办?用库改变大小即可

import Image
     ##更改图片大小
     infile = “C:\\Users\\Administrator\\Desktop\\result\\商品货号:" + str(info_id[i]) + ".png“
     outfile = ”C:\\Users\\Administrator\\Desktop\\result1\\商品货号:" + str(info_id[i]) + ".png”
     im = Image.open(infile)
     (x, y) = im.size
     x_s = 80  ## 设置长
     y_s = 100  ## 设置宽
     out = im.resize((x_s, y_s), Image.ANTIALIAS)
     out.save(outfile,'png',quality = 95)

将大图片和小图片放在了result和result1两个不同的文件夹,需要再前边创建这两个文件夹:

if os.path.exists(r'C:\\Users\\Administrator\\Desktop\\result'): # 建立一个文件夹在桌面,文件夹为result
     print('result文件夹已经在桌面存在,继续运行程序……')
   else:
     print('result文件夹不在桌面,新建文件夹result')
     os.mkdir(r'C:\\Users\\Administrator\\Desktop\\result')
     print('文件夹建立成功,继续运行程序')
 
   if os.path.exists(r'C:\\Users\\Administrator\\Desktop\\result1'): # 建立一个文件夹在C盘,文件夹为result1
     print('result1文件夹已经在桌面存在,继续运行程序……')
   else:
     print('result1文件夹不在桌面,新建文件夹result1')
     os.mkdir(r'C:\\Users\\Administrator\\Desktop\\result1')
     print('文件夹建立成功,继续运行程序')

最后插入图片到excel中:

worksheet.insert_image(i + 1, len(info_first) + 1, 'C:\\Users\\Administrator\\Desktop\\result1\\' + "商品货号:" + str(info_id[i]) + '.png') ##写入图片
     time.sleep(1)##防止写入太快电脑死机
     plt.close() #  一定要关掉图片,不然python打开图片20个后会崩溃
 
   workbook.close()#最后关闭excel

得到的效果如下:

python批量制作雷达图的实现方法

附上完整代码:

import numpy as np
  import matplotlib.pyplot as plt
  import xlsxwriter
  from openpyxl import load_workbook
  import os
  import time
  from PIL import Image
  
  if __name__ == '__main__':
 
   if os.path.exists(r'C:\\Users\\Administrator\\Desktop\\result'): # 建立一个文件夹在桌面,文件夹为result
     print('result文件夹已经在桌面存在,继续运行程序……')
   else:
     print('result文件夹不在桌面,新建文件夹result')
     os.mkdir(r'C:\\Users\\Administrator\\Desktop\\result')
     print('文件夹建立成功,继续运行程序')
 
   if os.path.exists(r'C:\\Users\\Administrator\\Desktop\\result1'): # 建立一个文件夹在C盘,文件夹为result1
     print('result1文件夹已经在桌面存在,继续运行程序……')
   else:
     print('result1文件夹不在桌面,新建文件夹result1')
     os.mkdir(r'C:\\Users\\Administrator\\Desktop\\result1')
     print('文件夹建立成功,继续运行程序')
 
   wb = load_workbook(filename=r'C:\Users\Administrator\Desktop\数据指标.xlsx') ##读取路径
   ws = wb.get_sheet_by_name("Sheet1") ##读取名字为Sheet1的sheet表
 
   info_id = []
   info_first = []
 
   for row_A in range(2, 32): ## 遍历第2行到32行
     id = ws.cell(row=row_A, column=1).value ## 遍历第2行到32行,第1列
     info_id.append(id)
   for col in range(2, 9): ##读取第1到9列
     first = ws.cell(row=1, column=col).value
     info_first.append(first) ##得到1到8列的标签
   print(info_id)
   print(info_first)
 
   info_data = []
   for row_num_BtoU in range(2, len(info_id) + 2): ## 遍历第2行到32行
     row_empty = [] ##建立一个空数组作为临时储存地,每次换行就被清空
     for i in range(2, 9): ## 遍历第2行到32行,第2到9列
       data_excel = ws.cell(row=row_num_BtoU, column=i).value
       if data_excel == None:
         pass
       else:
         row_empty.append(data_excel) ##将单元格信息储存进去
     info_data.append(row_empty)
   print(info_data)
   print(len(info_data))
 
   # 设置雷达各个顶点的名称
   labels = np.array(info_first)
   # 数据个数
   data_len = len(info_first)
   # 新建一个excel保存结果
   workbook = xlsxwriter.Workbook('C:\\Users\\Administrator\\Desktop\\result.xlsx')
   worksheet = workbook.add_worksheet() # 创建一个工作表对象
   # 字体格式
   font = workbook.add_format(
     {'border': 1, 'align': 'center', 'font_size': 11, 'font_name': '微软雅黑'}) ##字体居中,11号,微软雅黑,给一般的信息用的
   # 写下第一行第一列的标签
   worksheet.write(0, 0, '商品货号', font)
   ##设置图片的那一列宽度
   worksheet.set_column(0, len(info_first) + 1, 11) # 设定第len(info_first) + 1列的宽度为11
 
   ##写入标签
   for k in range(0, 7):
     worksheet.write(0, k + 1, info_first[k], font)
   # 写入最后一列标签
   worksheet.write(0, len(info_first) + 1, '雷达图', font)
 
   # 将其他参数写入excel中
   for j in range(0, len(info_id)):
     worksheet.write(j + 1, 0, info_id[j], font) # 写入商品货号
     worksheet.set_row(j, 76) ##设置行宽
     for x in range(0, len(info_first)):
       worksheet.write(j + 1, x + 1, info_data[j][x], font) # 写入商品的其他参数
 
   for i in range(0, len(info_id)):
     data = np.array(info_data[i])
 
     angles = np.linspace(0, 2 * np.pi, data_len, endpoint=False)
     data = np.concatenate((data, [data[0]])) # 闭合
     angles = np.concatenate((angles, [angles[0]])) # 闭合
 
     fig = plt.figure()
     ax = fig.add_subplot(111, polar=True) # polar参数!!
     ax.plot(angles, data, 'bo-', linewidth=2) # 画线
     ax.fill(angles, data, facecolor='r', alpha=0.25) # 填充
     ax.set_thetagrids(angles * 180 / np.pi, labels, fontproperties="SimHei")
     ax.set_title("商品货号:" + str(info_id[i]), va='bottom', fontproperties="SimHei")
     ax.set_rlim(3.8, 5) # 设置雷达图的范围
     ax.grid(True)
     plt.savefig("C:\\Users\\Administrator\\Desktop\\result\\商品货号:" + str(info_id[i]) + ".png", dpi=120)
     # plt.show()在python中显示
 
     ##更改图片大小
     infile = "C:\\Users\\Administrator\\Desktop\\result\\商品货号:" + str(info_id[i]) + ".png"
     outfile = "C:\\Users\\Administrator\\Desktop\\result1\\商品货号:" + str(info_id[i]) + ".png"
     im = Image.open(infile)
     (x, y) = im.size
     x_s = 80 ## 设置长
     y_s = 100 ## 设置宽
     out = im.resize((x_s, y_s), Image.ANTIALIAS)
     out.save(outfile, 'png', quality=95)
 
     worksheet.insert_image(i + 1, len(info_first) + 1,
                'C:\\Users\\Administrator\\Desktop\\result1\\' + "商品货号:" + str(
                  info_id[i]) + '.png') ##写入图片
     time.sleep(1) ##防止写入太快电脑死机
     plt.close() # 一定要关掉图片,不然python打开图片20个后会崩溃
 
   workbook.close() # 最后关闭excel

以上就是本文介绍利用python批量制作雷达图的实现方法,希望给学习python的大家有所帮助

Python 相关文章推荐
python使用pyhook监控键盘并实现切换歌曲的功能
Jul 18 Python
python爬虫headers设置后无效的解决方法
Oct 21 Python
Pandas 数据框增、删、改、查、去重、抽样基本操作方法
Apr 12 Python
解决Pycharm中import时无法识别自己写的程序方法
May 18 Python
Python实现的微信好友数据分析功能示例
Jun 21 Python
使用python对文件中的数值进行累加的实例
Nov 28 Python
pytorch中的自定义数据处理详解
Jan 06 Python
Python通过TensorFLow进行线性模型训练原理与实现方法详解
Jan 15 Python
Python中包的用法及安装
Feb 11 Python
python GUI库图形界面开发之pyinstaller打包python程序为exe安装文件
Feb 26 Python
Matplotlib 绘制饼图解决文字重叠的方法
Jul 24 Python
python dict如何定义
Sep 02 Python
python 添加用户设置密码并发邮件给root用户
Jul 25 #Python
Python文件夹与文件的相关操作(推荐)
Jul 25 #Python
浅谈python类属性的访问、设置和删除方法
Jul 25 #Python
python直接访问私有属性的简单方法
Jul 25 #Python
python类:class创建、数据方法属性及访问控制详解
Jul 25 #Python
python实现汉诺塔方法汇总
Jul 25 #Python
python魔法方法-属性访问控制详解
Jul 25 #Python
You might like
综合图片计数器
2006/10/09 PHP
我用php+mysql写的留言本
2006/10/09 PHP
php加密解密函数authcode的用法详细解析
2013/10/28 PHP
Thinkphp5.0框架的Db操作实例分析【连接、增删改查、链式操作等】
2019/10/11 PHP
浅谈laravel5.5 belongsToMany自身的正确用法
2019/10/17 PHP
Javascript 类与静态类的实现
2010/04/01 Javascript
Whatever:hover 无需javascript让IE支持丰富伪类
2010/06/29 Javascript
js导出table数据到excel即导出为EXCEL文档的方法
2013/10/10 Javascript
JS实现一个按钮的方法
2015/02/05 Javascript
JavaScript实现数字数组按照倒序排列的方法
2015/04/06 Javascript
PHP和NodeJs开发的应用如何共用Session
2015/04/16 NodeJs
jQuery常用的一些技巧汇总
2016/03/26 Javascript
Angular 应用技巧总结
2016/09/14 Javascript
前端开发必知的15个jQuery小技巧
2017/01/22 Javascript
Angular 通过注入 $location 获取与修改当前页面URL的实例
2017/05/31 Javascript
JS加密插件CryptoJS实现AES加密操作示例
2018/08/16 Javascript
基于vue通用表单解决方案的思考与分析
2019/03/16 Javascript
详解微信小程序实现跑马灯效果(附完整代码)
2019/04/29 Javascript
微信小程序的开发范式BeautyWe.js入门详解
2019/07/10 Javascript
在vue中阻止浏览器后退的实例
2019/11/06 Javascript
使用konva和vue-konva库实现拖拽滑块验证功能
2020/04/27 Javascript
js仿京东放大镜效果
2020/08/09 Javascript
JS highcharts实现动态曲线代码示例
2020/10/16 Javascript
[15:58]DOTA2国际邀请赛采访专栏:Tongfu.Sansheng&KingJ,DK.rOtk
2013/08/08 DOTA
简单谈谈python中的Queue与多进程
2016/08/25 Python
Python实现将HTML转换成doc格式文件的方法示例
2017/11/20 Python
Python操作MySQL模拟银行转账
2018/03/12 Python
解决Python获取字典dict中不存在的值时出错问题
2018/10/17 Python
python使用socket 先读取长度,在读取报文内容示例
2019/09/26 Python
Python字符串hashlib加密模块使用案例
2020/03/10 Python
解决reload(sys)后print失效的问题
2020/04/25 Python
解决IDEA 的 plugins 搜不到任何的插件问题
2020/05/04 Python
Python实现CAN报文转换工具教程
2020/05/05 Python
python 三种方法提取pdf中的图片
2021/02/07 Python
法律七进实施方案
2014/03/15 职场文书
ssh服务器拒绝了密码 请再试一次已解决(亲测有效)
2022/08/14 Servers