python - 在 Pycharm 中使用 sys.stdin 从文件中读取

Question

我正在尝试测试一个使用 Pycharm 逐行读取文件的简单代码。

for line in sys.stdin:
    name, _ = line.strip().split("\t")
    print name

我在同一个目录下有我要输入的文件:lib.txt

如何使用输入文件在 Pycharm 中调试我的代码？

Answer 1

如果您使用 fileinput 模块而不是尝试直接读取 stdin，则可以解决此问题。

对于文件输入，如果脚本在参数中接收到文件名，它将按顺序从参数中读取。在您的情况下，将上面的代码替换为:

import fileinput
for line in fileinput.input():
    name, _ = line.strip().split("\t")
    print name

fileinput 的伟大之处在于，如果没有提供参数(或者如果提供了参数“-”)，它默认为 stdin。

现在您可以创建运行配置并提供要用作标准输入的文件的文件名作为脚本的唯一参数。

阅读更多关于 fileinput here 的信息