python批量制作雷達圖的實現(xiàn)方法
前言
因為工作需要有時候要畫雷達圖,但是數(shù)據(jù)好多組怎么辦?不能一個一個點excel去畫吧,那么可以利用python進行批量制作,得到樣式如下:

首先制作一個演示的excel,評分為excel隨機數(shù)生成:
1 =INT((RAND()+4)*10)/10
加入標簽等得到的excel樣式如下(部分,共計32行):

那么接下來就是打開python寫碼了,本文是基于pycharm進行編寫
wb = load_workbook(filename=r'C:\Users\Administrator\Desktop\數(shù)據(jù)指標.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 = [] ##建立一個空數(shù)組作為臨時儲存地,每次換行就被清空
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\數(shù)據(jù)指標.xlsx') ##讀取路徑
ws = wb.get_sheet_by_name("Sheet1") ##讀取名字為Sheet1的sheet表
需要用到庫:
import xlsxwriter
from openpyxl import load_workbook
在命令指示符下輸入:
pip install xlsxwriter
等待安裝即可,后面的庫也是如此:

將第一列ID儲存,以及第一行的標簽,標簽下面的數(shù)值分別儲存在:
info_id = [] info_first = [] info_data = []
讀取數(shù)據(jù)后接下來需要設(shè)置寫入的格式:
workbook = xlsxwriter.Workbook('C:\\Users\\Administrator\\Desktop\\result.xlsx')
worksheet = workbook.add_worksheet() # 創(chuàng)建一個工作表對象
#字體格式
font = workbook.add_format(
{'border': 1, 'align': 'center', 'font_size': 11, 'font_name': '微軟雅黑'}) ##字體居中,11號,微軟雅黑,給一般的信息用的
#寫下第一行第一列的標簽
worksheet.write(0, 0, '商品貨號', font)
##設(shè)置圖片的那一列寬度
worksheet.set_column(0, len(info_first) + 1, 11) # 設(shè)定第len(info_first) + 1列的寬度為11
將標簽數(shù)據(jù)等寫入新的excel表格中:
#新建一個excel保存結(jié)果
workbook = xlsxwriter.Workbook('C:\\Users\\Administrator\\Desktop\\result.xlsx')
worksheet = workbook.add_worksheet() # 創(chuàng)建一個工作表對象
#字體格式
font = workbook.add_format(
{'border': 1, 'align': 'center', 'font_size': 11, 'font_name': '微軟雅黑'}) ##字體居中,11號,微軟雅黑,給一般的信息用的
#寫下第一行第一列的標簽
worksheet.write(0, 0, '商品貨號', font)
##設(shè)置圖片的那一列寬度
worksheet.set_column(0, len(info_first) + 1, 11) # 設(shè)定第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)
制作雷達圖:
#設(shè)置雷達各個頂點的名稱
labels = np.array(info_first)
#數(shù)據(jù)個數(shù)
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參數(shù)?。?
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)# 設(shè)置雷達圖的范圍
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 ## 設(shè)置長
y_s = 100 ## 設(shè)置寬
out = im.resize((x_s, y_s), Image.ANTIALIAS)
out.save(outfile,'png',quality = 95)
將大圖片和小圖片放在了result和result1兩個不同的文件夾,需要再前邊創(chuàng)建這兩個文件夾:
if os.path.exists(r'C:\\Users\\Administrator\\Desktop\\result'): # 建立一個文件夾在桌面,文件夾為result
print('result文件夾已經(jīng)在桌面存在,繼續(xù)運行程序……')
else:
print('result文件夾不在桌面,新建文件夾result')
os.mkdir(r'C:\\Users\\Administrator\\Desktop\\result')
print('文件夾建立成功,繼續(xù)運行程序')
if os.path.exists(r'C:\\Users\\Administrator\\Desktop\\result1'): # 建立一個文件夾在C盤,文件夾為result1
print('result1文件夾已經(jīng)在桌面存在,繼續(xù)運行程序……')
else:
print('result1文件夾不在桌面,新建文件夾result1')
os.mkdir(r'C:\\Users\\Administrator\\Desktop\\result1')
print('文件夾建立成功,繼續(xù)運行程序')
最后插入圖片到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() # 一定要關(guān)掉圖片,不然python打開圖片20個后會崩潰
workbook.close()#最后關(guān)閉excel
得到的效果如下:

附上完整代碼:
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文件夾已經(jīng)在桌面存在,繼續(xù)運行程序……')
else:
print('result文件夾不在桌面,新建文件夾result')
os.mkdir(r'C:\\Users\\Administrator\\Desktop\\result')
print('文件夾建立成功,繼續(xù)運行程序')
if os.path.exists(r'C:\\Users\\Administrator\\Desktop\\result1'): # 建立一個文件夾在C盤,文件夾為result1
print('result1文件夾已經(jīng)在桌面存在,繼續(xù)運行程序……')
else:
print('result1文件夾不在桌面,新建文件夾result1')
os.mkdir(r'C:\\Users\\Administrator\\Desktop\\result1')
print('文件夾建立成功,繼續(xù)運行程序')
wb = load_workbook(filename=r'C:\Users\Administrator\Desktop\數(shù)據(jù)指標.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 = [] ##建立一個空數(shù)組作為臨時儲存地,每次換行就被清空
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))
# 設(shè)置雷達各個頂點的名稱
labels = np.array(info_first)
# 數(shù)據(jù)個數(shù)
data_len = len(info_first)
# 新建一個excel保存結(jié)果
workbook = xlsxwriter.Workbook('C:\\Users\\Administrator\\Desktop\\result.xlsx')
worksheet = workbook.add_worksheet() # 創(chuàng)建一個工作表對象
# 字體格式
font = workbook.add_format(
{'border': 1, 'align': 'center', 'font_size': 11, 'font_name': '微軟雅黑'}) ##字體居中,11號,微軟雅黑,給一般的信息用的
# 寫下第一行第一列的標簽
worksheet.write(0, 0, '商品貨號', font)
##設(shè)置圖片的那一列寬度
worksheet.set_column(0, len(info_first) + 1, 11) # 設(shè)定第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)
# 將其他參數(shù)寫入excel中
for j in range(0, len(info_id)):
worksheet.write(j + 1, 0, info_id[j], font) # 寫入商品貨號
worksheet.set_row(j, 76) ##設(shè)置行寬
for x in range(0, len(info_first)):
worksheet.write(j + 1, x + 1, info_data[j][x], font) # 寫入商品的其他參數(shù)
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參數(shù)??!
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) # 設(shè)置雷達圖的范圍
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 ## 設(shè)置長
y_s = 100 ## 設(shè)置寬
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() # 一定要關(guān)掉圖片,不然python打開圖片20個后會崩潰
workbook.close() # 最后關(guān)閉excel
以上就是本文介紹利用python批量制作雷達圖的實現(xiàn)方法,希望給學(xué)習(xí)python的大家有所幫助
相關(guān)文章
用Python代碼自動生成文獻的IEEE引用格式的實現(xiàn)
這篇文章主要介紹了用Python代碼自動生成文獻的IEEE引用格式的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03

