Python for循环满足条件时仅单次打开文件的实现方法

直接在循环外部初始化文件句柄标记，在条件触发时增加一次打开状态判断即可，不需要二次遍历循环，也不需要使用break中断流程，完全兼容原有代码逻辑。

# 循环外初始化文件句柄为None，作为未打开的标记
f = None
for i in range(large_number):
    # 保留原有计算a_i的全部逻辑
    # compute a_i
    if a_i is condition:
        # 仅在文件从未打开过时执行打开操作，后续再触发条件不会重复open
        if f is None:
            f = open('file.log', 'wb')
        # 保留原位置的所有后续计算逻辑，不需要调整位置、不需要break
        # more computations proceeding

# 循环结束后如果文件打开过，记得关闭释放资源
if f is not None:
    f.close()

方案说明 #

• 整个逻辑只需要遍历一次循环区间，不存在额外遍历的性能开销，也不需要把循环内的计算逻辑拆到循环外
• 第一次触发判定条件时文件就会被打开，后续所有触发条件的场景都会跳过打开操作，不会出现重复打开文件的问题
• 原代码中open的第二个参数wb需要写成字符串格式'wb'，直接写wb会被Python识别为变量，触发未定义报错
• 如果需要在文件操作逻辑中统一使用上下文管理器写法避免漏关文件，也可以用空上下文做占位实现：

from contextlib import nullcontext

# 用空上下文占位，未触发文件打开时调用文件相关方法也不会报错
f = nullcontext()
for i in range(large_number):
    # compute a_i
    if a_i is condition:
        if isinstance(f, nullcontext):
            f = open('file.log', 'wb')
        # 后续所有计算、写入逻辑正常写即可，不需要额外判断文件是否存在
        # 例如 f.write(log_content)

# 最后关闭真实打开的文件
if not isinstance(f, nullcontext):
    f.close()

内容的提问来源于stack exchange，提问作者SKPS