Python使用字典记录迭代索引实现续接运算的问题排查

代码存在两处核心bug，直接导致第二轮迭代无输出：

• 迭代范围计算逻辑错误：后续迭代时直接将用户新输入的迭代次数n作为range终点，首次迭代到索引5后，第二轮start值为6，用户输入n=5时，range(6, 5+1)是空区间，for循环不会执行任何逻辑。实际上新输入的n是本轮需要新增的迭代次数，range终点应该为start + n。
• 运算底数被错误覆盖：第二轮分支中x = sorted(num_dic.values())[-1]会把底数x替换为上一轮最后一次的运算结果（首次跑完x会被改成32），但需求是始终对最初输入的同一个变量做迭代运算，该赋值逻辑完全错误。

额外优化点：原代码用字符串作为字典key，每次读取最后一个索引都要排序、类型转换，直接用整数作为key更简洁高效。

def function(x, i):
    return x**i


num_dic = {}
current_index = 0  # 记录已迭代到的索引位置
base_x = None  # 存储固定的运算底数，仅首次运行时赋值

while True:
    # 首次运行时初始化底数
    if base_x is None:
        n = int(input("number of iterations: "))
        base_x = int(input("Number to perform operation: "))
    else:
        n = int(input("number of iterations: "))
    
    # 计算本轮迭代的起止区间
    start = current_index + 1
    end = start + n
    # 执行迭代运算
    for i in range(start, end):
        total = function(base_x, i)
        print(f"{base_x} ** {i} = {total}")
        num_dic[i] = total
        current_index = i  # 更新已迭代到的位置
    
    # 询问用户是否继续
    while True:
        ask = input("Do you want to continue?(Y to continue/ N to exit): ")
        if ask.upper() in ("Y", "N"):
            break
        print("Please enter a correct operation (Y/N) ")
    
    if ask.upper() == "N":
        break

输入第一次迭代次数5、底数2，选择继续后输入第二次迭代次数5，输出完全符合预期：

number of iterations: 5
Number to perform operation: 2
2 ** 1 = 2
2 ** 2 = 4
2 ** 3 = 8
2 ** 4 = 16
2 ** 5 = 32
Do you want to continue?(Y to continue/ N to exit): y
number of iterations: 5
2 ** 6 = 64
2 ** 7 = 128
2 ** 8 = 256
2 ** 9 = 512
2 ** 10 = 1024
Do you want to continue?(Y to continue/ N to exit): n

内容的提问来源于stack exchange，提问作者right