ChatGPT解决这个技术问题 Extra ChatGPT

如何使用 open with 语句打开文件

我正在研究如何在 Python 中进行文件输入和输出。我编写了以下代码来将文件中的名称列表(每行一个)读取到另一个文件中,同时根据文件中的名称检查名称并将文本附加到文件中的出现处。该代码有效。能不能做得更好?

我想对输入和输出文件使用 with open(... 语句,但看不到它们如何位于同一个块中,这意味着我需要将名称存储在临时位置。

def filter(txt, oldfile, newfile):
    '''\
    Read a list of names from a file line by line into an output file.
    If a line begins with a particular name, insert a string of text
    after the name before appending the line to the output file.
    '''

    outfile = open(newfile, 'w')
    with open(oldfile, 'r', encoding='utf-8') as infile:
        for line in infile:
            if line.startswith(txt):
                line = line[0:len(txt)] + ' - Truly a great person!\n'
            outfile.write(line)

    outfile.close()
    return # Do I gain anything by including this?

# input the name you want to check against
text = input('Please enter the name of a great person: ')    
letsgo = filter(text,'Spanish', 'Spanish2')
“意味着我需要将名称存储在临时位置”?你能解释一下你的意思吗?
请注意,filter()a built-in function,因此您可能应该为您的函数选择一个不同的名称。
@Tom 命名空间中的函数是否覆盖了内置函数?
@UpTide:是的,Python 以 LEGB 顺序运行——本地、封闭、全局、内置(参见 stackoverflow.com/questions/291978/…)。所以,如果你做一个全局函数(filter()),它会在内置的filter()之前找到

s
steveha

Python 允许将多个 open() 语句放在一个 with 中。你用逗号分隔它们。您的代码将是:

def filter(txt, oldfile, newfile):
    '''\
    Read a list of names from a file line by line into an output file.
    If a line begins with a particular name, insert a string of text
    after the name before appending the line to the output file.
    '''

    with open(newfile, 'w') as outfile, open(oldfile, 'r', encoding='utf-8') as infile:
        for line in infile:
            if line.startswith(txt):
                line = line[0:len(txt)] + ' - Truly a great person!\n'
            outfile.write(line)

# input the name you want to check against
text = input('Please enter the name of a great person: ')    
letsgo = filter(text,'Spanish', 'Spanish2')

不,通过在函数末尾放置显式 return 不会获得任何收益。您可以使用 return 提前退出,但最后有它,并且函数将在没有它的情况下退出。 (当然对于返回值的函数,您可以使用 return 来指定要返回的值。)

在引入 with 语句时,Python 2.5 或 Python 2.6 不支持将多个 open() 项与 with 一起使用,但在 Python 2.7 和 Python 3.1 或更高版本中受支持。

http://docs.python.org/reference/compound_stmts.html#the-with-statement http://docs.python.org/release/3.1/reference/compound_stmts.html#the-with-statement

如果您正在编写必须在 Python 2.5、2.6 或 3.0 中运行的代码,请将 with 语句嵌套为建议的其他答案或使用 contextlib.nested


M
MrWonderful

使用这样的嵌套块,

with open(newfile, 'w') as outfile:
    with open(oldfile, 'r', encoding='utf-8') as infile:
        # your logic goes right here

C
Community

你可以用块嵌套你的。像这样:

with open(newfile, 'w') as outfile:
    with open(oldfile, 'r', encoding='utf-8') as infile:
        for line in infile:
            if line.startswith(txt):
                line = line[0:len(txt)] + ' - Truly a great person!\n'
            outfile.write(line)

这比您的版本要好,因为您保证即使您的代码遇到异常也会关闭 outfile。显然,您可以使用 try/finally 来做到这一点,但 with 是做到这一点的正确方法。

或者,正如我刚刚了解到的,您可以在一个 with 语句中拥有多个上下文管理器作为 described by @steveha。在我看来,这似乎是比嵌套更好的选择。

对于您的最后一个小问题,回报没有真正的目的。我会删除它。


b
brother-bilo

有时,您可能希望打开数量不定的文件并将每个文件一视同仁,您可以使用 contextlib

from contextlib import ExitStack
filenames = [file1.txt, file2.txt, file3.txt]

with open('outfile.txt', 'a') as outfile:
    with ExitStack() as stack:
        file_pointers = [stack.enter_context(open(file, 'r')) for file in filenames]                
            for fp in file_pointers:
                outfile.write(fp.read())