文件在达到末尾之前已关闭 - 使用Python中的Exitstack

Question

我已经使用以下代码同时读取多个文件

from contextlib import ExitStack

files_to_parse = [file1, file2, file3]
    
with ExitStack() as stack:
    files = [stack.enter_context(open(i, "r")) for i in files_to_parse]
    for rows in zip(*files):
        for r in rows:
            #do stuff
        

，但是我注意到，由于我所有的文件都没有相同数量的行，只要最短文件到达末尾，所有文件都会关闭。

我使用了上面的代码（在Stackoverflow上找到了它），因为我需要同时解析几个文件（以节省时间）。这样做，将计算时间除以4。但是，所有文件并非完全是因为我上面提到的问题。

有什么方法可以解决这个问题吗？

Answer 1

open 可以用作上下文管理器，但不必。您可能会以古老的方式使用它，在这里您负责关闭，也就是说

try:
    files = [open(fname) for fname in filenames]
    # here do what you need to with files
finally:
    for file in files:
        file.close()