数据可视化2---3D坐标轴

￡神魔★判官ぃ 2022-06-08 13:18 310阅读 0赞

import matplotlib.pyplot as plt
    import numpy as np
    import matplotlib as mpl
    from mpl_toolkits.mplot3d import Axes3D
    import traceback
        
    BJ = dict()
    SH = dict()
    GZ = dict()
    SZ = dict()
    ZB = dict() 
    BJ_num = dict()
    SH_num = dict()
    GZ_num = dict()
    SZ_num = dict()
    ZB_num = dict()
    
    hlist = list()
    lbj = list()
    lsh = list()
    lsz = list()
    lgz = list()
    lzb = list()
    
    def init():
        ulist = ['数据库管理员','网络工程师',
                     '网络管理员','系统工程师',
                     '信息主管','硬件工程师',
                     'ERP工程师','ERP实施顾问',
                     'erp实施顾问',
                     '软件实施工程师','实施工程师',
                     '项目实施工程师','数据库工程师',
                     '高级网络工程师','运维工程师']
        try:
            for i in range(len(ulist)):
                pos = ulist[i]
                BJ[pos]=SH[pos]=GZ[pos]=SZ[pos]=ZB[pos]=0
                BJ_num[pos]=SH_num[pos]=GZ_num[pos]=SZ_num[pos]=ZB_num[pos]=0
                hlist.append(pos)
        except:
            traceback.print_exc()
            print('ERROR1')
    
    def BeiJing(pos,salary,area):
        try:
            BJ[pos] += salary
            BJ_num[pos] += 1
        except:
            print('ERROR2')
            
    def ShangHai(pos,salary,area):
        try:
            SH[pos] += salary
            SH_num[pos] += 1
        except:
            print('ERROR3')
            
    def GuangZhou(pos,salary,area):
        try:
            GZ[pos] += salary
            GZ_num[pos] += 1
        except:
            print('ERROR4')
            
    def ShenZhen(pos,salary,area):
        try:
            SZ[pos] += salary
            SZ_num[pos] += 1
        except:
            print('ERROR5')
            
    def ZiBo(pos,salary,area):
        try:
            ZB[pos] += salary
            ZB_num[pos] += 1
        except:
            print('ERROR6')
    
    def chance(pos,salary,area):
        try:
            if(area == '北京'):
                BeiJing(pos,salary,area)
            elif(area == '广州'):
                ShangHai(pos,salary,area)
            elif(area == '深圳'):
                GuangZhou(pos,salary,area)
            elif(area == '上海'):
                ShenZhen(pos,salary,area)
            elif(area == '淄博'):
                ZiBo(pos,salary,area)
        except:
            print('ERROR7')
    
    def search(pos):
        for i in range(len(hlist)):
            u = hlist[i]
            if(u == pos):
                j = 1
                break
            else:
                continue
        return j
        
    def getData(count):
        num = 0
        fpath = "D:\数据\数据可视化\data.txt"
        with open(fpath,'r',encoding='utf-8') as f:
            for i in range(count):
                try:
                    line = f.readline()
                    pos = eval(line.split(',')[0])
                    pos = pos.split('（')[0]
                    pos = pos.split('-')[0]
                    salary = eval(line.split(',')[2])
                    if(salary == '面议'):
                        continue
                    salary = salary.split('-')[1]
                    salary = int(salary)
                    area = eval(line.split(',')[3])
                    area = area.split('-')[0]
                    flag = search(pos)
                    if(flag == 1):
                        num = num + 1
                        chance(pos,salary,area)
                    else:
                        continue
                except:
                    continue
    
    def updateData():
        BJ['ERP实施顾问']+=BJ['erp实施顾问']
        BJ_num['ERP实施顾问']+=BJ_num['erp实施顾问']
        SH['ERP实施顾问']+=SH['erp实施顾问']
        SH_num['ERP实施顾问']+=SH_num['erp实施顾问']
        GZ['ERP实施顾问']+=GZ['erp实施顾问']
        GZ_num['ERP实施顾问']+=GZ_num['erp实施顾问']
        SZ['ERP实施顾问']+=SZ['erp实施顾问']
        SZ_num['ERP实施顾问']+=SZ_num['erp实施顾问']
        ZB['ERP实施顾问']+=ZB['erp实施顾问']
        ZB_num['ERP实施顾问']+=ZB_num['erp实施顾问']
        try:
            for k,v in BJ.items():
                if(BJ_num[k]!=0):
                    v = int(v/BJ_num[k])
                else:
                    v = 0
                lbj.append(v)
            for k,v in SZ.items():
                if(SZ_num[k]!=0):
                    v = int(v/SZ_num[k])
                else:
                    v = 0
                lsz.append(v)    
            for k,v in GZ.items():
                if(GZ_num[k]!=0):
                    v = int(v/GZ_num[k])
                else:
                    v = 0
                lgz.append(v)
            for k,v in SH.items():
                if(SH_num[k]!=0):
                    v = int(v/SH_num[k])
                else:
                    v = 0
                lsh.append(v)
            for k,v in ZB.items():
                if(ZB_num[k]!=0):
                    v = int(v/ZB_num[k])
                else:
                    v = 0
                lzb.append(v)
            lbj.pop(8)
            lsh.pop(8)
            lsz.pop(8)
            lgz.pop(8)
            lzb.pop(8)
            hlist.pop(8)
        except:
            print('ERROR8')
    
            
    def draw_figure():
        wlist = [lbj,lsz,lgz,lsh,lzb]
        warea = ['北京','深圳','广州','上海','淄博']
        
        mpl.rcParams['font.size'] = 10
        mpl.rcParams['font.family'] = 'simHei'
        font = {'color':'orange',
                'size':12}
        font1 = {'color':'orange',
                'size':15}
        font2 = {'color':'cyan',
                'size':20}
        
        fig = plt.figure(figsize=(20,15))
        ax = fig.add_subplot(111,projection='3d')
        
        ind = np.arange(len(hlist))
        x = ind
        c = ['red','green','blue','yellow','purple']
        for i in range(5):
            y = wlist[i]
            ax.bar(x,y,i,zdir='y',color=c[i],alpha=0.8)
        ax.set_xticks(ind)
        ax.set_xticklabels(hlist,rotation=80)
        ax.set_yticks(np.arange(5))
        ax.set_yticklabels(warea,rotation=-10,fontdict = font)
        ax.set_xlabel('      职 业',fontdict = font1)
        ax.set_ylabel('    地区',fontdict = font1)
        ax.set_zlabel('工资',fontdict = font1)
        ax.set_title('各个地区对应职位的\'最高\'薪水统计',fontdict = font2)
        plt.savefig("D:\数据\hua.png")
        plt.show()
        
    def main():
        init()
        getData(15710)
        updateData()
        draw_figure()
    main()

![Center][]

[Center]: /images/20220608/101db2c7c7954b4cac78282ea0907ffb.png