怎么优化代码(在用户名已创建时，用户重新输入)

Question

请问在当用户输入要注册的用户名显示已被占用时，怎么优化代码可以重新回到输入用户名的那个步骤?

Answer 1

user_names = ['zhan','wang','li','lin']
reg_name = input('请输入你要注册的用户名: ')
while True:
    if reg_name in user_names:
        reg_name = input('该用户名已注册，请重新输入: ')
    else:
        user_names.append(reg_name)
        print('用户名注册成功')
        break