本文共 410 字,大约阅读时间需要 1 分钟。
每当我写个脚本涉及文件输入时,一般写法都是下面这种
import sysfile_in = sys.argv[1]for line in open(file_in, 'r'): commands
这个操作有一个缺点就是,如果我想从管道里面传入数据给Python的话,就会报错,因为原代码要求文件而不是标准输入。
这个问题可以通过Python的一个标准库: 进行解决。
import fileinputfor line in fileinput.input(): process(line)
fileinput.intpu()
会帮我们自动处理输入。如果sys.argv[1:]
里有输入文件,它就会对里面所有的文件进行遍历,如果sys.argv
为空,那么它就会从标准输入sys.stdin
里读取输入,如果输入文件的文件名是"-", 同样地会从标准输入里读取输入。这样子就省去了我们自己写条件语句进行判断输入类型。
转载地址:http://ooqwm.baihongyu.com/