如何读取stdin？

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

当前回答

以下是学习Python的内容：

import sys
data = sys.stdin.readlines()
print "Counted", len(data), "lines."

在Unix上，您可以通过以下方式进行测试：

% cat countlines.py | python countlines.py 
Counted 3 lines.

在Windows或DOS上，您可以执行以下操作：

C:\> type countlines.py | python countlines.py 
Counted 3 lines.

其他回答

我非常惊讶，到目前为止没有人提到过这个黑客：

python -c "import sys; set(map(sys.stdout.write,sys.stdin))"

在python2中，您可以放弃set（）调用，但这两种方法都可以

import sys

for line in sys.stdin:
    print(line)

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

当使用-c命令时，作为一种巧妙的方法，您可以将shell脚本命令放在以$符号开头的括号内的引号中，而不是读取stdin（在某些情况下更灵活）。

e.g.

python3 -c "import sys; print(len(sys.argv[1].split('\n')))" "$(cat ~/.goldendict/history)"

这将统计goldendict历史文件中的行数。

其他人提出的答案：

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。

您可以从stdin读取数据，然后将输入存储到“数据”中，如下所示：

data = ""
for line in sys.stdin:
    data += line

推荐文章