Python中获取重定向输入本质是读取sys.stdin,当命令行重定向时(如python script.py < input.txt),sys.stdin自动关联重定向源。

Python 中获取重定向输入,本质是读取标准输入(sys.stdin),它在命令行中被重定向时(如 python script.py )会自动从文件读取,无需额外配置。
使用 sys.stdin 读取重定向输入
这是最直接、最常用的方式。当程序通过 重定向输入时,<code>sys.stdin 就指向那个输入源(文件或管道),你可以像读文件一样操作它:
- 用
sys.stdin.read()一次性读取全部内容(适合小到中等输入) - 用
sys.stdin.readline()逐行读取(内存友好,推荐用于大输入或未知长度) - 用
for line in sys.stdin:迭代读取(简洁、Pythonic)
兼容键盘输入和重定向输入
如果希望脚本既能处理重定向输入(python a.py ),也能交互式地从键盘输入(<code>python a.py),可以检测 sys.stdin.isatty():
-
True表示连接的是终端(键盘输入) -
False表示已被重定向(来自文件/管道)
例如:
立即学习“Python免费学习笔记(深入)”;
import sysif not sys.stdin.isatty():
data = sys.stdin.read().strip()
else:
data = input("请输入:")
注意缓冲与换行符
重定向输入时,sys.stdin 默认是行缓冲(终端下)或全缓冲(重定向时),但对用户代码透明;真正需留意的是换行符:
-
readline()和迭代方式都会保留每行末尾的\n,通常需要.strip()或.rstrip('\n') -
input()在交互模式下自动去掉末尾换行,但在重定向时不可用(会抛EOFError)
替代方案:用 argparse 显式接收文件路径
如果你更倾向显式控制输入来源(而非依赖 shell 重定向),可用 argparse 接收文件名参数,并打开读取:
parser = argparse.ArgumentParser()
parser.add_argument('file', nargs='?', type=argparse.FileType('r'), default=sys.stdin)
args = parser.parse_args()
for line in args.file: ...
这样既支持 python a.py input.txt,也支持 python a.py < input.txt(此时 default=sys.stdin 生效)。











