python3函数读取文件写入文件默认覆盖文件

Question

我想创建一个函数，读入一个txt文件，删除每一行的前导空格和尾随空格，然后写入一个文件，默认覆盖我读入的文件，但可以选择写入一个新文件. 这是我的代码。

def cleanfile(inputfile, outputfile = inputfile):
    file1 = open(inputfile,'r')
    file2 = open(outputfile, 'w')
    lines = list(file1)
    newlines = map(lambda x: x.strip(), lines)
    newlines = list(newlines)
    for i in range(len(newlines)):
        file2.write(newlines[i] + '\n')
    file1.close()
    file2.close()    
cleanfile('hw.txt',)
cleanfile('hw.txt','hw_2.txt')

但它给了我错误。 NameError: name 'inputfile' 没有定义

请问如何解决这个问题并实现我的目标？非常感谢。

Answer 1

Python 中的标准约定是使用 None 作为默认值并检查它。

def cleanfile(inputfile, outputfile = None):
    if outputfile is None:
        outputfile = inputfile
    file1 = open(inputfile,'r')
    file2 = open(outputfile, 'w')
    lines = list(file1)
    newlines = map(lambda x: x.strip(), lines)
    newlines = list(newlines)
    for i in range(len(newlines)):
        file2.write(newlines[i] + '\n')
    file1.close()
    file2.close()    
cleanfile('hw.txt',)
cleanfile('hw.txt','hw_2.txt')

Answer 2

您不能将 outputfile=inputfile 设置为默认参数。这是 Python 的一个限制 - 当指定默认参数时，'inputfile' 不作为变量存在。

您可以使用标记值：

sentinel = object()
def func(argA, argB=sentinel):
    if argB is sentinel:
       argB = argA
    print (argA, argB)

func("bar")           # Prints 'bar bar'
func("bar", None)     # Prints 'bar None'