python 使用matlab库绘制雷达图（包含读取处理excel数据）

谁践踏了优雅 2022-10-14 00:40 201阅读 0赞

import numpy as np
    import pandas as pd
    import matplotlib.pyplot as plt
    import cairosvg
    
    from openpyxl import load_workbook
    import  time
    def plot_radar(data,lables,id):
        '''
        the first column of the data is the cluster name;
        the second column is the number of each cluster;
        the last are those to describe the center of each cluster.
        '''
    
    
        # labels = data.iloc[:, 2:].columns
        # centers = pd.concat([data.iloc[:, 2:], data.iloc[:, 2]], axis=1)
        lables=["","","","","","","","",""]
        for ele in range(len(id)):
            plt.rcParams["font.sans-serif"] = ["KaiTi"]
            # plt.rcParams['font.family'] = 'YaHei Consolas Hybrid'  # 设置字体样式
            plt.rcParams['font.size'] = '15'  # 设置字体大小
            plt.rcParams["axes.unicode_minus"] = False
    
            centers=[data[ele]]
            for i in range(len(centers[0])):
                # centers[0][i
                centers[0][i]/=10
            centers[0].append(centers[0][0])
    
            centers = np.array(centers)
    
            n = 8
            angles = np.linspace(0, 2 * np.pi, n, endpoint=False)
    
            angles = np.concatenate((angles, [angles[0]]))
            # np.sort(-angles)
            #print(angles)
    
            angles=angles[::-1]
            fig = plt.figure()
            ax = fig.add_subplot(111,polar=True)  # 设置坐标为极坐标
            fig.patch.set_alpha(0.)  # 或者其他值
            # 画若干个五边形
            floor = np.floor(centers.min())  # 大于最小值的最大整数
            ceil = np.ceil(centers.max())  # 小于最大值的最小整数
            for i in np.arange(floor, ceil + 0.2, 0.2):
                ax.plot(angles, [i] * (n+1), '-', lw=0.5, color='#00000080')
            #print(angles)
            # 画不同客户群的分割线
            for i in range(n):
                ax.plot([angles[i], angles[i]], [floor, ceil], ' ', lw=0.5, color='black')
    
            # 画不同的客户群所占的大小
            kinds=[1]
            print(ele+1,"个")
           # print(angles)
            #print(len(lables))
            # print(centers[i])
            # ax.plot(angles, centers[i], lw=2, label=kinds[i])
            for i in range(len(kinds)):
                ax.plot(angles, centers[i], lw=2,color='#f5d731')
                ax.fill(angles, centers[i],color ='#f5d731C9')
                #print(lables.append(lables[0]))
                ax.set_thetagrids(angles * 180 / np.pi, lables)  # 设置显示的角度，将弧度转换为角度
               # plt.legend(loc='lower right', bbox_to_anchor=(1.5, 0.0))  # 设置图例的位置，在画布外
                # print(floor)
    
            ax.set_theta_zero_location('N')  # 设置极坐标的起点（即0°）在正北方向，即相当于坐标轴逆时针旋转90°
            ax.spines['polar'].set_visible(False)  # 不显示极坐标最外圈的圆
            ax.grid(False)  # 不显示默认的分割线
            ax.set_yticks([])  # 不显示坐标间隔
            # plt.savefig('files3/res.png')
            fig.savefig('files2/{}.svg'.format(id[ele]), transparent=True)
            cairosvg.svg2png(url=r"files2/{}.svg".format(id[ele]), write_to="files/{}.png".format(id[ele]))
            #time.sleep(1)
    
            plt.close(fig)
            #if ele>2:exit()
            # plt.show()
    
    wb = load_workbook(filename='WFA决赛评分报告数据.xlsx')  ##读取路径
    ws = wb.get_sheet_by_name("Sheet1")  ##读取名字为Sheet1的sheet表
    
    info_id = []
    info_first = []
    info_data = []
    
    print(ws.max_row)
    #
    for row_A in range(2, ws.max_row+1):  ## 遍历
        id = ws.cell(row=row_A, column=2).value  ##作品名称
    
        code=ws.cell(row=row_A, column=1).value
    
        id=str(code)+"-"+id
    
        if len(id)>25:
            id=id[:25]
        # id=id.replace(" ","-")
    
        info_id.append(id)
        #print(id)
    
    for row_num_BtoU in range(2, len(info_id) + 2):  ## 遍历
        row_empty = []  ##建立一个空数组作为临时储存地，每次换行就被清空
        for i in range(4, 12):  ## 遍历第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)
    
    
    lables=['产品定位','    信息传达与\n    外观设计','     包装功能性\n     与产品体验','       研发理念与\n       创新亮点','原料与工艺','营养与安全'
            ,'颜色与气味     ','口味与口感      ','后味与整体感受     ']
    plot_radar(info_data,lables,info_id)
    # label=['gg','hhh','yyy']
    # print(label[len(label)::-1])