Python 等价于 Perl 的 while (<>) {...}?

Question

我编写了很多逐行处理文件的小脚本。在 Perl 中，我使用

while (<>) {
    do stuff;
}

这很方便，因为它不关心输入来自哪里(文件或标准输入)。

在 Python 中我使用这个

if len(sys.argv) == 2: # there's a command line argument
    sys.stdin = file(sys.argv[1])
for line in sys.stdin.readlines():
    do stuff

这似乎不是很优雅。是否有 Python 习语可以轻松处理文件/stdin 输入？

Answer 1

标准库中的fileinput模块正是你想要的:

import fileinput

for line in fileinput.input(): ...