Python：如何在try-except的except块获取try中已输入的变量x

你的问题出在：当仅输入一个值时，x,n=input().split()会因为拆分后只有1个元素，无法解包给两个变量而直接抛出ValueError，此时x和n都没有被成功定义，所以后续执行print(root(x,n))时就会触发NameError。

要解决这个问题，你需要先获取拆分后的输入列表，再根据列表长度分别处理，而不是直接强制解包成两个变量。下面是两种可行的修改方案：

方案1：直接判断输入长度（更直观） #

def root(base, exp=2):   
    return base ** (1 / exp)

# 先获取拆分后的输入列表
input_parts = input("Enter the base and nth_root :").split()

if len(input_parts) == 2:
    # 输入两个值，转换为整数后调用函数
    x, n = int(input_parts[0]), int(input_parts[1])
    print(root(x, n))
elif len(input_parts) == 1:
    # 输入单个值，使用默认的n=2
    x = int(input_parts[0])
    print("By default, power is 0.5")
    print(root(x))
else:
    # 处理输入过多的情况
    print("Invalid input: please enter either one number (base) or two numbers (base and root)")

方案2：结合try-except处理（保留你原来的异常处理思路） #

def root(base, exp=2):   
    return base ** (1 / exp)

try:
    input_parts = input("Enter the base and nth_root :").split()
    if len(input_parts) == 2:
        x, n = int(input_parts[0]), int(input_parts[1])
        result = root(x, n)
    elif len(input_parts) == 1:
        x = int(input_parts[0])
        print("By default, power is 0.5")
        result = root(x)
    else:
        # 输入超过两个值时主动抛出异常
        raise ValueError("Please enter at most two numbers")
    
    print(result)
except ValueError as e:
    # 处理转换失败或输入格式错误的情况
    print(f"Input error: {e}")

两种方案都能避免NameError的问题，核心思路是不要在未确认输入长度的情况下直接解包多个变量，而是先拆分输入，再根据长度分支处理。

内容的提问来源于stack exchange，提问作者218 Mani kumar