英文词汇量测试

import numpy as np
from numpy import cov,corrcoef
import matplotlib.pyplot as plt
import astropy.units as u
from scipy.optimize import curve_fit
import scipy
import pandas as pd
import tkinter as tk
%matplotlib inline

vocabulary=pd.read_excel('./vocabulary20000.xlsx',header=None)#20200 words in vocabulary
vocabulary.columns=['num','words','a','b']
num=vocabulary['num']
words=vocabulary['words']

time=50 #自定义测试次数，次数越多更准确哦
init_num=5000#自定义初始词汇量，根据自身情况设置

class Application(tk.Frame):
    def __init__(self, master=None,result=0):
        tk.Frame.__init__(self, master)
        self.pack()
        self.createWidgets()


    def createWidgets(self):

        self.word=tk.Button(self)
        self.word["text"]="%s"%(words[numoftest])
        self.word.pack(side="top")

        self.hi_yes = tk.Button(self)
        self.hi_yes["text"] = "Known\n(click me)"
        self.hi_yes["command"] = self.say_yes
        self.hi_yes.pack(side="left")
        self.hi_no = tk.Button(self)
        self.hi_no["text"] = "UnKnown\n(click me)"
        self.hi_no["command"] = self.say_no
        self.hi_no.pack(side="right")


    def say_yes(self):
    #result=1
        global has_been_called
        has_been_called=True
        #return 1
        #print("oh! You get it")
        root.destroy()
    def say_no(self):
        global has_been_called
        has_been_called=False
        #result=0
        #return 0
        #print("em,try more ")
        root.destroy()


has_been_called=False
results=np.zeros((time,2))
numoftest=init_num+np.random.randint(1,50)
#numoftestorigin=1000
#result=0

for i in np.arange(time):
    root = tk.Tk()
    #root.withdraw()
    root.resizable(False,False)
    root.geometry("400x300+10+10")
    root.title('词汇量测试%d/%d'%(i,time))
#width=root.winfo_screenmmwidth
#height=root.winfo_screenheight
#root.deiconify()
    app = Application(master=root)
    app.mainloop()

    if has_been_called:
        results[i]=np.array([numoftest,1])
        numoftestorigin=numoftest
        numoftest=np.around((numoftest+numoftestorigin)/2+np.random.randint(10,100)*np.random.randint(1,10)+np.random.randint(10,50))
        numoftest=min(numoftest,19500+np.random.randint(1,100))
    else:
        results[i]=np.array([numoftest,0])
        numoftestorigin=numoftest
        numoftest=np.around((numoftest+numoftestorigin)/2-np.random.randint(10,100)*np.random.randint(1,10)-np.random.randint(10,50))
        numoftest=max(numoftest,3000+np.random.randint(1,100))
        #print(results[0:time])

wordnum=results[:,0]
yesorno=results[:,1]
correct=np.sum(yesorno==1)
correctrate=correct/time
correctrateper=format(correctrate,'.2%')
vocabularynum=np.around(np.mean(wordnum[-10:]))
plt.plot(wordnum,'.')
plt.title("Your correctness rate is %s "%correctrateper)
plt.xlabel('time of test')
plt.ylabel('number of words you know')
plt.savefig('vocabularyresult.png',dpi=400)
print("Your vocabulary is %s"%vocabularynum)