python3从命令终端获取参数并打印输入

Question

lines = []
for line in fileinput.input():
    lines.append(line)
    print(line, end='')

我的问题是，如何让2进入程序，同时打印出input.txt的内容呢？到目前为止，它会得到 2 作为文件名并尝试将其作为文件打开，这与预期不同。你在图片中看到。

Answer 1

可以使用 python 内部的 sys.stdin 文件句柄访问 input.txt 文件的内容。必须使用 sys.argv 数组访问参数。

这是一个示例脚本。

#!/usr/bin/python
import sys
arg=sys.argv[1] #take first argument
lines=[]
for line in sys.stdin:
  lines.append(line) # save line from stdin to the array. Watch out for memory usage.


print arg
print lines