python 如何获得重定向输入

13次阅读

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