本文介绍两种简洁、pythonic的方式重定义内置input函数，使其每次调用时从预设列表（或字符串块）中顺序返回一个值，适用于测试、算法调试等场景，无需全局计数器或yield协程。
在编写需要交互输入的脚本（如算法题、CLI工具或教学示例）时，我们常需在测试阶段绕过真实用户输入，改用一组预设值进行自动化验证。虽然可以用全局变量+闭包模拟，但更优雅、符合Python风格的做法是利用迭代器协议——因为 input() 本质是一个无参、单次调用即返回一个值的函数，这与迭代器的 __next__() 方法行为高度一致。
✅ 推荐方案一：用 iter(...).__next__ 替换 input（最推荐）
# 将 input 重定义为从列表顺序取值的迭代器
raw_inputs = ['8', '2', '1']
input = iter(raw_inputs).__next__
# 此后所有 input() 调用将依次返回 '8' → '2' → '1'
for _ in range(3):
print(input()) # 输出: 8, 2, 1✅ 优势：
- 简洁、不可变、线程安全（无状态）；
- 完全复用 Python 迭代器机制，语义清晰；
- 若需保留原始列表用于其他逻辑，可直接引用 raw_inputs；
- 自动抛出 StopIteration（对应真实 input() 在 EOF 时的 EOFError），便于边界测试。
⚠️ 注意：若调用次数超过列表长度，会触发 StopIteration 异常。如需静默忽略或提供默认值，可简单封装：
from itertools import chain input = chain(['8', '2', '1'], ['']).__next__ # 超出后返回空字符串
✅ 推荐方案二：用 list.pop()（适合逆序使用）
# 注意：pop 默认从末尾移除，因此需倒序初始化或使用 pop(0)
inputs = ['1', '2', '8'] # 倒序写，或用 inputs = ['8','2','1'][::-1]
input = inputs.pop
for _ in range(3):
print(input()) # 输出: 1 → 2 → 8（若 inputs 为 ['1','2','8']）⚠️ 注意：此方式会原地修改列表，且 pop() 不是纯函数；若需多次运行测试，每次需重新初始化列表。
立即学习“Python免费学习笔记（深入）”；
? 进阶技巧：处理多行输入（如 OJ 题目样例）
实际刷题或解析输入块时，常遇到多行文本。此时可结合 str.splitlines() 构建迭代器：
# 模拟标准输入中的多行数据 input_data = '''\ 8 2 1 ''' input = iter(input_data.splitlines()).__next__ print(input()) # '8' print(input()) # '2' print(input()) # '1'
? 提示：splitlines(keepends=False) 自动处理 \n, \r\n, \r 等换行符，比手动 split('\n') 更健壮。
? 总结
- 首选 iter([...]).__next__：零副作用、高可读性、完全符合 Python 迭代器协议；
- 避免 global 计数器或手写 yield 协程——input() 不是生成器，强行用 yield 反而增加复杂度（需调用 next() 或管理生成器状态）；
- 所有方案均保持类型一致性：原生 input() 返回 str，故预设值也建议用字符串（如 ['8','2','1']），避免类型混淆；
- 在正式项目中，建议通过 unittest.mock.patch 替换 input 更规范；但在脚本/学习/快速验证场景下，上述赋值法高效直接。
