隨機生成六位驗證碼函數版(python)

import random
def code(n=6,alpha=True):
    s = '' # 創建字符串變量,存儲生成的驗證碼
    for i in range(n):  # 通過for循環控制驗證碼位數
        num = random.randint(0,9)  # 生成隨機數字0-9
        if alpha: # 需要字母驗證碼,不用傳參,如果不需要字母的,關鍵字alpha=False
            upper_alpha = chr(random.randint(65,90))
            lower_alpha = chr(random.randint(97,122))
            num = random.choice([num,upper_alpha,lower_alpha])
        s = s + str(num)
    return s
print(code(6,False))  # 打印6位數字驗證碼
print(code(6,True))   # 打印6位數字字母混合驗證碼
print(code(4,False))  # 打印4位數字驗證碼
print(code(4,True))   # 打印4位數字字母混合驗證碼

以上代碼僅供參考,有問題可以留言,相互交流!