import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.pyplot import axis

def get_985(jd_tmp,tmp):
    jd_985=[]
    for i in range(len(jd_tmp)):
        if(tmp[i]=='√'):
            jd_985.append(jd_tmp[i])
    return jd_985

def get_school_dict(school_set,school_value):
    dict_school={}
    for i in range(0,len(school_set)):
        dict_school[unicode(school_set[i])]=school_value
    return dict_school
    
def get_value_list(datalist,dict_211,dict_211_value,dict_985,dict_985_value):
    stulist=[]
    for i in range(0,len(datalist)):
        tmp=0
        if(dict_211.get(unicode(datalist[i]))==1):
            if(dict_985.get(unicode(datalist[i]))==2):
                tmp=dict_985_value
            else:
                tmp=dict_211_value 
        else:
            tmp=0
        stulist.append(tmp)
        
    stulist.reverse()
    return stulist
    
studata=pd.read_excel(r'e:/test/src.xls')
judgedata=pd.read_excel(r'e:/test/judge.xlsx')
jd_211=judgedata[u'院校名稱'].tolist()
jd_985=get_985(jd_211,judgedata[985])

dict_985=get_school_dict(jd_985,2)
dict_211=get_school_dict(jd_211,1)

datalist=studata[u'報考院校'].tolist()
stulist=get_value_list(datalist,dict_211,1,dict_985,2)

cnt_211=0
cnt_985=0
cnt=0
for i in range(len(stulist)):
    if(stulist[i]==2):
        cnt_985+=1
    elif(stulist[i]==1):
        cnt_211+=1
    elif(stulist[i]==0):
        cnt+=1
        
axis([0, 1600, -1, 3])
plt.title("985 school:%d,211 school:%d,others:%d" %(cnt_985,cnt_211,cnt))
plt.plot(range(0,len(stulist)),stulist,color='g',marker='o')
plt.savefig(r'e:/test/school.png')
plt.show()
 

資料來源是某院校的考研報名統計,985,211院校的名單表格從網上下載一份即可,統計了考生報名雙非,211,985的人數分佈,繪製成圖如下:(0表示雙非,1表示211,2表示985)