初学python之验证码 大小写不限制

# 四位验证码
code = ''
for i in range(4):
    s = 'QWERTYUIOPASDFGHJKLZCXVBNMqwertyuioplkjhgfdsazxcvbnm123456789'
    words = random.randint(0, len(s) - 1)
    code +=s[words]

print("您的验证码为：" + code)
choic = input("请输入验证码：\n")

if choic.lower() == code.lower() or choic.upper() == code.upper():
    print("正确")
else:
    print("错误")