本文介绍两种pythonic方式重定义内置input函数，使其按顺序返回预设列表中的值，避免使用全局变量和可变状态，兼顾简洁性、可读性与实用性。
在单元测试、算法调试或在线判题（如LeetCode、Codeforces）场景中，我们常需模拟用户输入——即让 input() 函数不再等待终端输入，而是从一组预设值中依次返回。虽然用全局计数器配合闭包能实现（如问题中所示），但这种方式破坏了纯函数性、引入了可变状态，且不够“Pythonic”。
更优雅、更符合Python惯用法的方案是利用迭代器协议，将预设数据转换为可消耗的迭代器，并将其 __next__ 方法直接赋给 input 名称。以下是两种主流实现：
✅ 方案一：使用 iter(...).__next__（推荐）
raw_data = ['8', '2', '1']
input = iter(raw_data).__next__
for _ in range(3):
print(input()) # 输出: 8, 2, 1- ✅ 优势：语义清晰（明确表达“逐个取值”）、线程安全（无共享状态）、支持任意可迭代对象（列表、生成器、文件行等）；
- ⚠️ 注意：若调用次数超过列表长度，会抛出 StopIteration 异常（与真实迭代器行为一致）。如需静默终止或提供默认值，可进一步封装：
from itertools import islice input = lambda: next(iter(['8','2','1']), '') # 超限时返回空字符串
✅ 方案二：使用 list.pop()（仅限列表，需逆序）
raw_data = ['8', '2', '1']
input = raw_data[::-1].pop # 翻转后从末尾弹出 → 实际按原序返回
for _ in range(3):
print(input()) # 输出: 8, 2, 1- ✅ 优势：代码极简；
- ❌ 缺陷：修改原列表（pop 是就地操作）、不可复用（列表被清空）、不适用于元组等不可变序列。
? 进阶技巧：处理多行输入块
实际调试中，输入常为多行文本（如题目样例）：
# 模拟标准输入流：每行一个输入值 input_data = '''\ 8 2 1''' input = iter(input_data.splitlines()).__next__ print(input()) # '8' print(input()) # '2' print(input()) # '1'
该模式天然支持换行符处理，且 splitlines() 自动剥离 \n，无需手动 strip()。
立即学习“Python免费学习笔记（深入）”；
⚠️ 重要注意事项
- 不要在生产代码中重定义 input：这会污染全局命名空间，影响依赖真实输入的模块；
- 仅用于脚本调试或测试环境：建议在 if __name__ == '__main__': 块内局部覆盖；
- 类型一致性：原生 input() 总是返回字符串，因此预设列表元素也应为字符串（如 ['8', '2']），而非整数 [8, 2]，否则可能引发类型错误；
-
可复位需求？ 若需多次重放输入序列，应封装为函数或类，例如：
def make_input_mock(data): it = iter(data) return lambda: next(it) input = make_input_mock(['8','2','1'])
综上，iter([...]).__next__ 是最推荐的方式——它简洁、健壮、符合Python设计哲学，真正实现了“用迭代器思维解决输入模拟问题”。
