本文主要是介绍「Python标准库:fileinput」优雅的读取文件,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
每当我写个脚本涉及文件输入时,一般写法都是下面这种
import sysfile_in = sys.argv[1]for line in open(file_in, 'r'):commands
这个操作有一个缺点就是,如果我想从管道里面传入数据给Python的话,就会报错,因为原代码要求文件而不是标准输入。
这个问题可以通过Python的一个标准库: fileinput进行解决。
import fileinputfor line in fileinput.input():process(line)
fileinput.intpu()
会帮我们自动处理输入。如果sys.argv[1:]
里有输入文件,它就会对里面所有的文件进行遍历,如果sys.argv
为空,那么它就会从标准输入sys.stdin
里读取输入,如果输入文件的文件名是"-", 同样地会从标准输入里读取输入。这样子就省去了我们自己写条件语句进行判断输入类型。
这篇关于「Python标准库:fileinput」优雅的读取文件的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!