Python从末尾取第N个元素的代码输入格式兼容问题排查

问题现象 #

• 输入为一行正整数+一行负数时（示例如下），原代码抛出ValueError：

  1 2 3 4 5
  -6
  

• 输入为逐个数字分行输入时（示例如下），程序可正常输出预期结果：

  1
  2
  3
  4
  5
  -6
  

• 自行修改后的代码存在逻辑错误：输入1 2 3 4 5和-6时输出-6，输入1 5 9 7 5和-3时输出-3，但预期输出应为9。

原代码问题根源 #

原代码的while循环中，每次直接调用int(input())转换输入内容。当输入一行包含多个数字的字符串（如"1 2 3 4 5"）时，字符串含空格无法直接转为整数，因此触发ValueError。

正确修改方案 #

需要兼容两种输入格式：每行单个数字，或一行多个数字。核心逻辑是读取每行内容后拆分处理所有数字，直到遇到负数为止。

修改后的完整代码：

def find_nth_from_end(numbers, n):
    if n > len(numbers):
        return -n
    else:
        return numbers[-n]

def main():
    numbers = []
    while True:
        line = input().strip()
        if not line:
            continue
        num_strs = line.split()
        for s in num_strs:
            num = int(s)
            if num < 0:
                n = -num
                result = find_nth_from_end(numbers, n)
                print(result)
                return
            numbers.append(num)

if __name__ == "__main__":
    main()

修改说明 #

1. 每次读取整行输入，拆分出所有数字字符串后逐个转换为整数
2. 遇到负数时立即计算结果并输出，直接终止程序，避免无效循环
3. 同时支持两种输入格式，无需区分输入方式

错误修改的问题点 #

你之前的修改错误地将整行转换得到的数字列表作为单个元素添加到numbers中，导致numbers变成嵌套列表（如[[1,2,3,4,5]]）。此时len(numbers)为1，当n=6时满足n>len(numbers)，返回-6；当n=3时同样满足条件，返回-3，完全不符合预期逻辑。

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