Python中遍历range时如何在for循环中回退一步？

问题分析 #

首先你的代码存在一个明显bug：random.randint(0, 2)会生成0、1、2三个整数，但你后续判断char == 3，这个条件永远不会触发，同时漏掉了char == 1的判断，导致数字永远不会被添加到密码里，需要先修正这个问题。

另外，你尝试通过i -= 1让for循环“回退”是行不通的——Python的for循环是遍历可迭代对象（比如range生成的序列），每次迭代的i是从序列里取出的下一个值，修改当前的i不会影响下一次迭代的取值，下一次依然会取序列里的下一个元素，所以这个操作完全无效。

解决方案 #

方案一：for循环内嵌套while循环重试 #

既然必须用for循环，我们可以在每次for循环迭代里，用while循环反复随机选择字符类型，直到选到还有剩余数量的类型为止，这样就不需要“回退”for循环了：

import random

# 定义字符集
letters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
symbols = '!@#$%^&*()'
numbers = '0123456789'

# 输入需求数量（可替换为用户输入）
letters_left = 3
numbers_left = 2
symbols_left = 2
final_string = ''

# 总循环次数等于所需字符总数
total_chars = letters_left + numbers_left + symbols_left
for _ in range(total_chars):
    selected = False
    while not selected:
        # 随机选择字符类型：0=字母，1=数字，2=符号
        char_type = random.randint(0, 2)
        
        if char_type == 0 and letters_left > 0:
            final_string += random.choice(letters)
            letters_left -= 1
            selected = True
        elif char_type == 1 and numbers_left > 0:
            final_string += random.choice(numbers)
            numbers_left -= 1
            selected = True
        elif char_type == 2 and symbols_left > 0:
            final_string += random.choice(symbols)
            symbols_left -= 1
            selected = True

print(final_string)

方案二：先构建字符池再打乱（更简洁高效） #

这种方法逻辑更清晰，先用for循环生成所有需要的字符，再一次性打乱顺序，完全符合需求，且代码更简洁：

import random

letters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
symbols = '!@#$%^&*()'
numbers = '0123456789'

letters_left = 3
numbers_left = 2
symbols_left = 2

# 用for循环构建初始字符列表
password_chars = []
for _ in range(letters_left):
    password_chars.append(random.choice(letters))
for _ in range(numbers_left):
    password_chars.append(random.choice(numbers))
for _ in range(symbols_left):
    password_chars.append(random.choice(symbols))

# 随机打乱顺序
random.shuffle(password_chars)
final_string = ''.join(password_chars)

print(final_string)

说明 #

方案二的效率更高，不需要反复重试随机选择，逻辑也更直观，推荐使用。如果必须保留原代码中“随机选择类型”的逻辑，方案一可以满足需求。

内容的提问来源于stack exchange，提问作者i-drink-dirty-code