python - 从 stdin python 读取

Question

我试图从第一个命令行参数中获取文件名输入，如果没有，则从 stdin 读取，但由于某种原因，我的 if 语句没有似乎正在工作。我已将其粘贴在下面。任何帮助将不胜感激。

import sys


filename=sys.argv[1]

if filename==None:
   filename=sys.stdin.readlines()

Answer 1

您应该检查 sys.argv 的长度在尝试对其建立索引之前:

import sys
if len(sys.argv) > 1:
  filename=sys.argv[1]
else:
  filename=sys.stdin.readline().strip()

sys.argv 中的第一个元素始终是 python 脚本的文件名，因此 > 1 .

[编辑]

扩展fixxxer关于is的回答与 == :

is返回True如果比较的变量指向同一个对象。这适用于 None ，如None是一个单例。 ==另一方面，返回 True如果比较的变量相等，即它们可以指向两个不同(但相等)的对象和 ==将产生True .