從excel中讀取表格值,然后繪制雷達(dá)圖
from time import clock
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/ASUS/Desktop/result'):? # 建立一個(gè)文件夾在桌面箕戳,文件夾為result
? ? ? ? print('result文件夾已經(jīng)在桌面存在某残,繼續(xù)運(yùn)行程序……')
? ? else:
? ? ? ? print('result文件夾不在桌面,新建文件夾result')
? ? ? ? os.mkdir(r'C:/Users/ASUS/Desktop/result')
? ? ? ? print('文件夾建立成功陵吸,繼續(xù)運(yùn)行程序')
? ? if os.path.exists(r'C:/Users/ASUS/Desktop/result1'):? # 建立一個(gè)文件夾在C盤(pán)玻墅,文件夾為result1
? ? ? ? print('result1文件夾已經(jīng)在桌面存在,繼續(xù)運(yùn)行程序……')
? ? else:
? ? ? ? print('result1文件夾不在桌面壮虫,新建文件夾result1')
? ? ? ? os.mkdir(r'C:/Users/ASUS/Desktop/result1')
? ? ? ? print('文件夾建立成功澳厢,繼續(xù)運(yùn)行程序')
? ? wb = load_workbook(filename=r'C:/Users/ASUS/Desktop/b班成績(jī).xlsx')? ##讀取路徑
? ? ws = wb.get_sheet_by_name("Sheet1")? ##讀取名字為Sheet1的sheet表
? ? info_id = []
? ? info_first = []
? ? for row_A in range(2, 11):? ## 遍歷第2行到11行
? ? ? ? id = ws.cell(row=row_A, column=1).value? ## 遍歷第2行到11行,第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列的標(biāo)簽
? ? print(info_id)
? ? print(info_first)
? ? info_data = []
? ? for row_num_BtoU in range(2, len(info_id) + 2):? ## 遍歷第2行到11行
? ? ? ? row_empty = []? ##建立一個(gè)空數(shù)組作為臨時(shí)儲(chǔ)存地,每次換行就被清空
? ? ? ? for i in range(2, 9):? ## 遍歷第2行到11行剩拢,第2到9列
? ? ? ? ? ? data_excel = ws.cell(row=row_num_BtoU, column=i).value
? ? ? ? ? ? if data_excel == None:
? ? ? ? ? ? ? ? pass
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? row_empty.append(data_excel)? ##將單元格信息儲(chǔ)存進(jìn)去
? ? ? ? info_data.append(row_empty)
? ? print(info_data)
? ? print(len(info_data))
? ? ? # 設(shè)置雷達(dá)各個(gè)頂點(diǎn)的名稱(chēng)
? ? labels = np.array(info_first)
? ? ? # 數(shù)據(jù)個(gè)數(shù)
? ? data_len = len(info_first)
? ? ? # 新建一個(gè)excel保存結(jié)果
? ? workbook = xlsxwriter.Workbook('C:/Users/ASUS/Desktop/result.xlsx')
? ? worksheet = workbook.add_worksheet()? # 創(chuàng)建一個(gè)工作表對(duì)象
? ? ? # 字體格式
? ? font = workbook.add_format(
? ? ? ? {'border': 1, 'align': 'center', 'font_size': 11, 'font_name': '微軟雅黑'})? ##字體居中线得,11號(hào),微軟雅黑,給一般的信息用的
? ? ? # 寫(xiě)下第一行第一列的標(biāo)簽
? ? worksheet.write(0, 0, '學(xué)生', font)
? ? ? ##設(shè)置圖片的那一列寬度
? ? worksheet.set_column(0, len(info_first) + 1, 11)? # 設(shè)定第len(info_first) + 1列的寬度為11
? ? ? ##寫(xiě)入標(biāo)簽
? ? for k in range(0, 7):
? ? ? ? worksheet.write(0, k + 1, info_first[k], font)
? ? ? # 寫(xiě)入最后一列標(biāo)簽
? ? worksheet.write(0, len(info_first) + 1, '雷達(dá)圖', font)
? ? ? # 將其他參數(shù)寫(xiě)入excel中
? ? for j in range(0, len(info_id)):
? ? ? ? worksheet.write(j + 1, 0, info_id[j], font)? # 寫(xiě)入學(xué)生號(hào)
? ? ? ? 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)? # 寫(xiě)入成績(jī)的其他參數(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ù)B惴觥框都!
? ? ? ? ax.plot(angles, data, 'bo-', linewidth=2)? # 畫(huà)線
? ? ? ? ax.fill(angles, data, facecolor='r', alpha=0.25)? # 填充
? ? ? ? ax.set_thetagrids(angles * 180 / np.pi, labels, fontproperties="SimHei")
? ? ? ? ax.set_title("學(xué)生:" + str(info_id[i]), va='bottom', fontproperties="SimHei")
? ? ? ? ax.set_rlim(0, 100)? # 設(shè)置雷達(dá)圖的范圍
? ? ? ? ax.grid(True)
? ? ? ? plt.savefig("C:/Users/ASUS/Desktop/result/學(xué)生:" + str(info_id[i]) + ".png", dpi=120)
? ? ? ? #在python中顯示
? ? ? ? #plt.show()
? ? ? ? ##更改圖片大小
? ? ? ? infile = "C:/Users/ASUS/Desktop/result/學(xué)生:" + str(info_id[i]) + ".png"
? ? ? ? outfile = "C:/Users/ASUS/Desktop/result1/學(xué)生:" + str(info_id[i]) + ".png"
? ? ? ? im = Image.open(infile)
? ? ? ? (x, y) = im.size
? ? ? ? x_s = 80? ## 設(shè)置長(zhǎng)
? ? ? ? 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/ASUS/Desktop/result1/' + "學(xué)生:" + str(info_id[i]) + '.png')? ##寫(xiě)入圖片
? ? ? ? time.sleep(1)? ##防止寫(xiě)入太快電腦死機(jī)
? ? ? ? plt.close()? # 一定要關(guān)掉圖片,不然python打開(kāi)圖片20個(gè)后會(huì)崩潰
? ? ? ? clock()
? ? workbook.close()? # 最后關(guān)閉excel
? ? print('運(yùn)行時(shí)間 %ss'%clock())
運(yùn)行截圖:生成了另外一張result的表