Python 練習 隨機生成4位驗證碼

import random
if __name__ =="__main__":    #四位數字字母驗證碼的生成
    checkcode="" #保存驗證碼的變量
    for i in range(4):
        index=random.randrange(0,4)  #生成一個0~3中的數
        if index!=i and index +1 !=i:
            checkcode +=chr(random.randint(97,122))  # 生成a~z中的一個小寫字母
        elif index +1==i:
            checkcode +=chr(random.randint(65,90) ) # 生成A~Z中的一個大寫字母
        else:
            checkcode +=str(random.randint(1,9))  # 數字1-9
    print(checkcode)