如何读取stdin？

如何读取stdin？一些代码高尔夫挑战需要使用stdin进行输入。

当前回答

从Python 3.8开始，您可以使用赋值表达式：

while (line := input()):
    print(line)

其他回答

import sys

for line in sys.stdin:
    print(line)

请注意，这将在末尾包含一个换行符。要删除末尾的换行符，请使用@brittohaloran所说的line.rstrip（）。

从Python 3.8开始，您可以使用赋值表达式：

while (line := input()):
    print(line)

其他人提出的答案：

for line in sys.stdin:
  print line

是非常简单和Python化的，但必须注意的是，脚本将等待EOF，然后开始迭代输入行。

这意味着tail-f error_log | myscript.py不会按预期处理行。

这种用例的正确脚本是：

while 1:
    try:
        line = sys.stdin.readline()
    except KeyboardInterrupt:
        break

    if not line:
        break

    print line

更新从评论中可以看出，在python2上可能只涉及缓冲区，因此在发出打印调用之前，您最终会等待缓冲区填充或EOF。

对于Python 3，这将是：

# Filename e.g. cat.py
import sys

for line in sys.stdin:
    print(line, end="")

这基本上是cat（1）的一种简单形式，因为它没有在每行后面添加一个换行符。您可以使用这个（在您使用chmod+x cat.py标记文件可执行文件之后），例如：

echo Hello | ./cat.py

使用文件输入模块：

import fileinput

for line in fileinput.input():
    pass

fileinput将循环通过输入中指定为命令行参数中给定的文件名的所有行，如果没有提供参数，则循环通过标准输入。

注意：行将包含尾随换行符；要删除它，请使用line.rstrip（）。

推荐文章